Home
last modified time | relevance | path

Searched defs:tensor_out (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_ones.cc85 auto tensor_out = infer_out->cast<tensor::TensorPtr>(); in TEST_P() local
Dtest_ops_zeros.cc85 auto tensor_out = infer_out->cast<tensor::TensorPtr>(); in TEST_P() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/
Dbounding_box.cc117 TensorPtr *tensor_out) { in CreateTensorFromBoundingBoxList()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dmaxpool_grad.cc67 Tensor *tensor_out = ctx.Input(kSecondInputIndex); in SpatialMaxPoolWithArgMaxHelper() local
279 Tensor *tensor_out = ctx.Input(kSecondInputIndex); in CheckMaxPoolGrad() local
Dfractional_max_pool_grad.cc74 const Tensor *tensor_out = ctx.Input(1); in DoCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dfractional_max_pool_grad_cpu_kernel.cc81 T *tensor_out = reinterpret_cast<T *>(inputs[1]->device_ptr()); in FractionalMaxPoolGradLaunch() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_cost.cc587 int64_t tensor_out = output_tensor_h * output_tensor_w * output_tensor_n * output_tensor_c; in GetOptimalStr() local
635 …int64_t tensor_out = static_cast<int64_t>(node.tensor_parm.tensor_shape.shape_h * node.tensor_parm… in GetMinCostIn() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/
Dkernel_exec.h262 auto tensor_out = mindspore::MSTensor(impl); in set_out_tensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/
Dutil.cc612 auto tensor_out = op.GetOutputDesc(output_name); in InferShapeRangeTwoInOneOutBroadcast() local