Searched defs:tensor_in (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | all_swap.cc | 60 …auto tensor_in = CheckAndConvertUtils::CheckArgsType(op_name, input_args, 0, kObjectTypeTensorType… in InferType() local
73 …auto tensor_in = CheckAndConvertUtils::CheckArgsType(op_name, input_args, 0, kObjectTypeTensorType… in InferShapeAndType() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/ops/ |
| D | test_int64_support.py | 29 def construct(self, inputs, tensor_in): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | maxpool_grad.cc | 65 Tensor *tensor_in = ctx.Input(kFirstInputIndex); in SpatialMaxPoolWithArgMaxHelper() local
278 Tensor *tensor_in = ctx.Input(kFirstInputIndex); in CheckMaxPoolGrad() local
|
| D | fractional_max_pool_grad.cc | 73 const Tensor *tensor_in = ctx.Input(0); in DoCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | fractional_max_pool_grad_cpu_kernel.cc | 79 T *tensor_in = reinterpret_cast<T *>(inputs[0]->device_ptr()); in FractionalMaxPoolGradLaunch() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_cost.cc | 565 int64_t tensor_in = input_tensor_h * input_tensor_w * input_tensor_n * input_tensor_c; in GetOptimalStr() local
626 …int64_t tensor_in = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_h * op.arguments[0].te… in GetMinCostIn() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/ |
| D | kernel_exec.h | 235 auto tensor_in = mindspore::MSTensor(impl); in set_in_tensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | comm_ops.py | 778 def __check__(self, tensor_in, send_size, recv_size): argument
|