Searched refs:tensor_in (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_int64_support.py | 29 def construct(self, inputs, tensor_in): argument
30 result = inputs[tensor_in]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_cost.cc | 237 int64_t tensor_in = input_tensor_h * input_tensor_w * input_tensor_n * input_tensor_c; in GetOptimalStr() local
278 cost_op.push_back(StrDimK(tensor_in) + CostRedis(node, node_name_to_strategy, in GetOptimalStr()
299 …int64_t tensor_in = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_h * op.arguments[0].te… in GetMinCostIn() local
315 cost_in.push_back(StrDimI(tensor_in, tensor_filter)); in GetMinCostIn()
316 cost_in.push_back(StrDimJ(tensor_in, tensor_filter)); in GetMinCostIn()
317 cost_in.push_back(StrDimK(tensor_in)); in GetMinCostIn()
318 cost_in.push_back(StrDimDI(tensor_in, tensor_out)); in GetMinCostIn()
319 cost_in.push_back(StrDimDJ(tensor_in, tensor_out)); in GetMinCostIn()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | comm_ops.py | 608 def __check__(self, tensor_in, send_size, recv_size): argument
609 validator.check_subclass("tensor_in", tensor_in['dtype'], mstype.tensor, self.name)
615 validator.check_equal_int(len(tensor_in['shape']), 2, "tensor_in", self.name)
619 out_shape = [-1] + [tensor_in['shape'][1]]
621 'dtype': tensor_in['dtype'],
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_others.cc | 402 auto tensor_in = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplAllSwap() local
403 MS_EXCEPTION_IF_NULL(tensor_in); in InferImplAllSwap()
404 MS_EXCEPTION_IF_NULL(tensor_in->shape()); in InferImplAllSwap()
405 auto tensor_in_shape = tensor_in->shape()->shape(); in InferImplAllSwap()
429 auto tensor_out = std::make_shared<AbstractTensor>(tensor_in->element(), in InferImplAllSwap()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_kernel.h | 240 auto tensor_in = mindspore::MSTensor(impl); in set_in_tensor() local
241 kernel_->set_input(tensor_in, index); in set_in_tensor()
|