pymoose.edsl.tracer module#

class pymoose.edsl.tracer.AstTracer(role_map=None)[source]#

Bases: object

get_fresh_name(prefix)[source]#
trace(expressions)[source]#
Return type

Computation

visit(expression)[source]#
visit_AbsExpression(abs_expression)[source]#
visit_AddNExpression(add_n_expression)[source]#
visit_ArgmaxExpression(argmax_expression)[source]#
visit_ArgumentExpression(argument_expression)[source]#
visit_AtLeast2DExpression(atleast_2d_expression)[source]#
visit_BinaryOpExpression(expression)[source]#
visit_CastExpression(cast_expression)[source]#
visit_ConcatenateExpression(concatenate_expression)[source]#
visit_ConstantExpression(constant_expression)[source]#
visit_DecryptExpression(decrypt_expression)[source]#
visit_ExpExpression(exp_expression)[source]#
visit_ExpandDimsExpression(expand_dims_expression)[source]#
visit_HostPlacementExpression(host_placement_expression)[source]#
visit_IdentityExpression(identity_expression)[source]#
visit_IndexAxisExpression(index_axis_expression)[source]#
visit_InverseExpression(inverse_expression)[source]#
visit_LoadExpression(load_expression)[source]#
visit_Log2Expression(log2_expression)[source]#
visit_LogExpression(log_expression)[source]#
visit_MaximumExpression(maximum_expression)[source]#
visit_MeanExpression(mean_expression)[source]#
visit_MirroredPlacementExpression(mirrored_placement_expression)[source]#
visit_MuxExpression(mux_expression)[source]#
visit_OnesExpression(ones_expression)[source]#
visit_OutputExpression(output_expression)[source]#
visit_ReluExpression(relu_expression)[source]#
visit_ReplicatedPlacementExpression(replicated_placement_expression)[source]#
visit_ReshapeExpression(reshape_expression)[source]#
visit_SaveExpression(save_expression)[source]#
visit_SelectExpression(select_expression)[source]#
visit_ShapeExpression(shape_expression)[source]#
visit_SigmoidExpression(exp_expression)[source]#
visit_SliceExpression(slice_expression)[source]#
visit_SoftmaxExpression(softmax_expression)[source]#
visit_SqrtExpression(exp_expression)[source]#
visit_SqueezeExpression(squeeze_expression)[source]#
visit_StridedSliceExpression(slice_expression)[source]#
visit_SumExpression(sum_expression)[source]#
visit_TransposeExpression(transpose_expression)[source]#
visit_ZerosExpression(zeros_expression)[source]#
visit_placement_expression(placement_expression)[source]#
pymoose.edsl.tracer.trace(abstract_computation)[source]#
pymoose.edsl.tracer.trace_and_compile(abstract_computation, compiler_passes=None)[source]#