Searched refs:tuple_len (Results 1 – 16 of 16) sorted by relevance
83 self.tuple_len = Primitive("tuple_len")93 for i in range(self.tuple_len(output)):98 if self.tuple_len(jvp_input) == 1:143 self.tuple_len = Primitive("tuple_len")149 if self.tuple_len(front_input) == 1:
451 def tuple_len(x): function453 tuple_len(a)
102 size_t tuple_len = arg->elements().size(); in InferImplStack() local114 for (size_t i = 1; i < tuple_len; ++i) { in InferImplStack()121 primitive->set_attr("N", MakeValue(SizeToLong(tuple_len))); in InferImplStack()129 (void)ret_shape.insert(ret_shape.begin() + axis_value, SizeToLong(tuple_len)); in InferImplStack()936 size_t tuple_len = 0; in InferImplConcat() local941 tuple_len = arg->elements().size(); in InferImplConcat()944 tuple_len = args_spec_list.size(); in InferImplConcat()956 primitive->set_attr("inputNums", MakeValue(SizeToLong(tuple_len))); in InferImplConcat()967 for (size_t i = 1; i < tuple_len; ++i) { in InferImplConcat()
47 length = F.tuple_len(shape)
99 if axis is None or F.tuple_len(axis) == 0:
199 ndim = F.tuple_len(shape)989 if F.tuple_len(F.shape(x)) > 1:
1063 if isinstance(axis, int) or (isinstance(axis, tuple) and F.tuple_len(axis) == 1):2405 ndim_out = F.tuple_len(shape)2416 ndim_orig = F.tuple_len(shape)2417 ndim_out = ndim_orig - F.tuple_len(axes)
80 length = F.tuple_len(shape)
105 length = F.tuple_len(shape)
83 length = F.tuple_len(shape)
133 length = F.tuple_len(shape)
181 tuple_len = Primitive("tuple_len") variable
61 length = F.tuple_len(shape)
515 input_nums = F.tuple_len(dout)
645 for i in range(F.tuple_len(next_params)):
77 length = F.tuple_len(shape)