Home
last modified time | relevance | path

Searched defs:indice_shape (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsparse_fill_empty_rows.cc104 …auto indice_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape())[kSh… in SparseFillEmptyRowsInferShape() local
138 …auto indice_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape())[kSh… in SparseFillEmptyRowsFrontendInferShape() local
Dsparse_to_dense.cc42 …auto indice_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetS… in SparseToDenseInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/
Dsparse_dense_cwise_operation_gpu_kernel.cc85 …std::vector<int64_t> indice_shape = std::vector<int64_t>(inputs[kIndex0]->GetDeviceShapeVector().b… in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dgather_v2_ds_fission.cc44 auto indice_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(origin_node, 1); in CheckInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Din_place_op_v2_cpu_kernel.cc71 auto indice_shape = inputs[kIndex1]->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_concat.cc249 auto indice_shape = indice->GetTensorShape(); in DoCompute() local