Home
last modified time | relevance | path

Searched defs:second_dim (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dembedding_dense_backward.cc32 auto second_dim = IsDynamicRank(grad_shape) ? abstract::Shape::kShapeDimAny : grad_shape.back(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dadjust_contrast_v2_gpu_kernel.cc35 constexpr int last_dim = 1, second_dim = 2, third_dim = 3; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_einsum_adjust.cc63 auto second_dim = second_dims.substr(second_dims.length() - DIMENSION_2D); in CheckCanConvertToMatmul() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dop_node.cc772 constexpr size_t second_dim = 1; in CalcGather() local
892 constexpr size_t second_dim = 1; in CalcConcat() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/
Dconfig_file_parser.cc107 size_t first_dim = kIndex0, second_dim = kIndex0; in CombineDynamicImageString() local