Home
last modified time | relevance | path

Searched defs:tensor_out_shape (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dall_swap.cc89 ShapeVector tensor_out_shape = {abstract::Shape::kShapeDimAny, tensor_in_shape[1]}; in InferShapeAndType() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dpattern_matcher.h749 ShapeVector tensor_out_shape = new_tensor_ptr->shape(); in CalcConstantTensors() local
816 auto tensor_out_shape = tensor_1_abstract->shape()->shape(); in GetNewTensor() local
828 auto tensor_out_shape = tensor_3_abstract->shape()->shape(); in GetNewTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dfractional_max_pool_grad.cc83 auto tensor_out_shape = tensor_out->GetTensorShape(); in DoCompute() local