Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsparse_fill_empty_rows.cc94 abstract::ShapePtr output_values_shape; in SparseFillEmptyRowsInferShape() local
128 abstract::ShapePtr output_values_shape; in SparseFillEmptyRowsFrontendInferShape() local
Ddensetodense_set_operation.cc91 abstract::ShapePtr output_values_shape = std::make_shared<abstract::Shape>(max_output_values_vec); in DenseToDenseSetOperationInferShape() local
157 abstract::ShapePtr output_values_shape = std::make_shared<abstract::Shape>(output_values_vec); in DenseToDenseSetOperationFrontendInferShape() local
Dsparse_split.cc70 …abstract::BaseShapePtr output_values_shape = std::make_shared<abstract::TensorShape>(output_values… in GetOutputShapes() local
Dsspaddmm.cc393 std::vector<int64_t> output_values_shape = {x2_indices_shape_right}; in GetOutputShapes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/
Dsparse_reorder_gpu_kernel.cc72 auto output_values_shape = outputs.at(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_slice.cc139 auto output_values_shape = output_values->GetTensorShape(); in Slice() local