Searched defs:tensor_in (Results 1 – 5 of 5) sorted by relevance
29 def construct(self, inputs, tensor_in): argument
237 int64_t tensor_in = input_tensor_h * input_tensor_w * input_tensor_n * input_tensor_c; in GetOptimalStr() local299 …int64_t tensor_in = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_h * op.arguments[0].te… in GetMinCostIn() local
240 auto tensor_in = mindspore::MSTensor(impl); in set_in_tensor() local
402 auto tensor_in = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplAllSwap() local
608 def __check__(self, tensor_in, send_size, recv_size): argument