Module Import alias identifier ImportFrom alias ImportFrom alias ImportFrom alias alias alias ImportFrom alias alias alias alias ImportFrom alias FunctionDef arguments args NameParam NameParam defaults body ImportFrom alias ImportFrom alias Assign NameStore Call AttributeLoad NameLoad attr NameLoad ImportFrom alias Assign NameStore Call NameLoad AttributeLoad AttributeLoad NameLoad attr attr Num Assign NameStore Call NameLoad AttributeLoad AttributeLoad NameLoad attr attr Num Assign NameStore Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr AttributeLoad AttributeLoad NameLoad attr attr Assign NameStore Call NameLoad NameLoad ListLoad NameLoad NameLoad Assert CompareEq AttributeLoad NameLoad attr NameLoad BinOpMod Str TupleLoad NameLoad AttributeLoad NameLoad attr Return NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults body If Call NameLoad AttributeLoad NameLoad attr NameLoad body Return Call NameLoad AttributeLoad NameLoad attr ListLoad NameLoad NameLoad keyword Call AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad NameLoad attr orelse If Call NameLoad AttributeLoad NameLoad attr NameLoad body Return Call NameLoad AttributeLoad NameLoad attr ListLoad NameLoad NameLoad keyword Call AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad NameLoad attr Assert Call NameLoad AttributeLoad NameLoad attr NameLoad BinOpMod Str TupleLoad NameLoad AttributeLoad NameLoad attr Assert Call NameLoad AttributeLoad NameLoad attr NameLoad BinOpMod Str TupleLoad NameLoad AttributeLoad NameLoad attr If BoolOpAnd CompareEq AttributeLoad AttributeLoad NameLoad attr attr Num CompareEq AttributeLoad AttributeLoad NameLoad attr attr Num body Assign NameStore Call NameLoad NameLoad NameLoad Return Call NameLoad keyword NameLoad keyword ListLoad NameLoad NameLoad keyword AttributeLoad NameLoad attr orelse If CompareEq AttributeLoad AttributeLoad NameLoad attr attr Num body Assign NameStore Call NameLoad NameLoad NameLoad Assert CompareEq AttributeLoad AttributeLoad NameLoad attr attr Num BinOpMod Str TupleLoad AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign NameStore Call NameLoad NameLoad TupleLoad NameLoad Assign NameStore Call NameLoad AttributeLoad NameLoad attr Num Return Call NameLoad keyword NameLoad keyword TupleLoad NameLoad keyword Num keyword NameLoad orelse If CompareEq AttributeLoad AttributeLoad NameLoad attr attr Num body Assert CompareEq AttributeLoad AttributeLoad NameLoad attr attr Num BinOpMod Str TupleLoad AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign NameStore Call NameLoad NameLoad NameLoad Assign NameStore Call NameLoad NameLoad TupleLoad NameLoad Assign NameStore Call NameLoad AttributeLoad NameLoad attr Num Return Call NameLoad keyword NameLoad keyword TupleLoad NameLoad keyword Num keyword NameLoad orelse Assert BoolOpAnd CompareEq AttributeLoad AttributeLoad NameLoad attr attr Num CompareEq AttributeLoad AttributeLoad NameLoad attr attr Num BinOpMod Str TupleLoad AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign NameStore Call NameLoad NameLoad NameLoad Assign NameStore Call NameLoad AttributeLoad NameLoad attr Num Return Call NameLoad keyword NameLoad keyword TupleLoad NameLoad NameLoad keyword TupleLoad Num Num keyword NameLoad decorator_list NameLoad FunctionDef arguments args NameParam NameParam defaults NameLoad body If CompareIs NameLoad NameLoad body Assign NameStore Num If Call NameLoad AttributeLoad NameLoad attr NameLoad body Return NameLoad Assert Call NameLoad AttributeLoad NameLoad attr NameLoad BinOpMod Str TupleLoad NameLoad AttributeLoad NameLoad attr Assert CompareEq AttributeLoad AttributeLoad NameLoad attr attr Num Call Str NameLoad AttributeLoad NameLoad attr If CompareEq NameLoad Num body Return Call AttributeLoad CompareNotEq NameLoad Num attr orelse If CompareEq NameLoad Num body Return Call AttributeLoad Call NameLoad NameLoad attr orelse If CompareEq NameLoad Num body Return Call AttributeLoad NameLoad attr Call AttributeLoad BinOpPow NameLoad Num attr orelse Return BinOpPow Call AttributeLoad BinOpPow Call NameLoad NameLoad NameLoad attr BinOpDiv Num NameLoad decorator_list NameLoad
N/A N/A numpy np N/A prims frontend typed_macro ndtypes ScalarT ArrayT make_array_type syntax PrimCall Call OuterMap Map syntax.helpers make_closure _get_vdot_fn N/A N/A a b N/A N/A reductions vdot frontend ast_conversion N/A vdot_untyped N/A N/A ast_conversion translate_function_value vdot type_inference specialize N/A vec_type_a N/A make_array_type N/A N/A a type elt_type 1 N/A vec_type_b N/A make_array_type N/A N/A b type elt_type 1 N/A result_scalar_type N/A N/A N/A N/A a type elt_type combine N/A N/A b type elt_type N/A vdot_typed N/A specialize vdot_untyped N/A vec_type_a vec_type_b N/A N/A N/A vdot_typed return_type result_scalar_type N/A Expected return type %s but got %s from vdot N/A result_scalar_type N/A vdot_typed return_type N/A vdot_typed N/A dot N/A N/A a b N/A N/A N/A N/A isinstance N/A a type ScalarT N/A N/A N/A PrimCall N/A prims multiply N/A a b type N/A N/A N/A b type combine N/A a type N/A N/A N/A isinstance N/A b type ScalarT N/A N/A N/A PrimCall N/A prims multiply N/A a b type N/A N/A N/A a type combine N/A b type N/A N/A isinstance N/A a type ArrayT N/A Expected %s to be array but got %s N/A a N/A a type N/A N/A isinstance N/A a type ArrayT N/A Expected %s to be array but got %s N/A b N/A b type N/A N/A N/A N/A N/A a type rank 1 N/A N/A N/A b type rank 1 N/A N/A vdot N/A _get_vdot_fn a b N/A N/A Call fn vdot args N/A a b type N/A vdot return_type N/A N/A N/A N/A N/A a type rank 1 N/A N/A vdot N/A _get_vdot_fn a b N/A N/A N/A N/A b type rank 2 N/A Don't know how to multiply %s and %s N/A N/A a type N/A b type N/A vdot_col N/A make_closure vdot N/A a N/A result_vec_type N/A make_array_type N/A vdot return_type 1 N/A N/A Map fn vdot_col args N/A b axis 1 type result_vec_type N/A N/A N/A N/A N/A b type rank 1 N/A N/A N/A N/A N/A a type rank 2 N/A Don't know how to multiply %s and %s N/A N/A a type N/A b type N/A vdot N/A _get_vdot_fn b a N/A vdot_row N/A make_closure vdot N/A b N/A result_vec_type N/A make_array_type N/A vdot return_type 1 N/A N/A Map fn vdot_row args N/A a axis 0 type result_vec_type N/A N/A N/A N/A N/A N/A a type rank 2 N/A N/A N/A b type rank 2 N/A Don't know how to multiply %s and %s N/A N/A a type N/A b type N/A vdot N/A _get_vdot_fn a b N/A result_matrix_type N/A make_array_type N/A vdot return_type 2 N/A N/A OuterMap fn vdot args N/A a b axis N/A 0 1 type result_matrix_type N/A typed_macro norm N/A N/A x ord N/A None N/A N/A N/A ord None N/A N/A ord 2 N/A N/A isinstance N/A x type ScalarT N/A N/A x N/A N/A isinstance N/A x type ArrayT N/A Argument to 'norm' must be array, got %s : %s N/A x N/A x type N/A N/A N/A N/A x type rank 1 N/A Norm currently only supported for vectors, not for %s : %s x N/A x type N/A N/A ord 0 N/A N/A N/A N/A N/A x 0 sum N/A N/A N/A ord 1 N/A N/A N/A N/A N/A abs x sum N/A N/A N/A ord 2 N/A N/A N/A N/A np sqrt N/A N/A N/A x 2 sum N/A N/A N/A N/A N/A N/A N/A abs x ord sum N/A 1.0 ord N/A typed_macro