| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/map_tensor/ |
| D | map_tensor_put_with_status_cpu_kernel.cc | 85 const auto &values_shape = inputs.at(kIndex2)->GetShapeVector(); in Resize() local
114 …StatusCpuKernelMod::InitSizeLists(const ShapeVector &keys_shape, const ShapeVector &values_shape) { in InitSizeLists()
|
| D | map_tensor_put_cpu_kernel.cc | 83 const auto &values_shape = inputs.at(kIndex2)->GetShapeVector(); in Resize() local
122 …sorPutCpuKernelMod::InitSizeLists(const ShapeVector &keys_shape, const ShapeVector &values_shape) { in InitSizeLists()
|
| D | map_tensor_get_data_cpu_kernel.cc | 79 const auto &values_shape = outputs.at(kIndex1)->GetShapeVector(); in Resize() local
106 …etDataCpuKernelMod::InitSizeLists(const ShapeVector &keys_shape, const ShapeVector &values_shape) { in InitSizeLists()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/map_tensor/ |
| D | map_tensor_put_gpu_kernel.cc | 77 const auto &values_shape = inputs.at(kIndex2)->GetShapeVector(); in Resize() local
104 …sorPutGpuKernelMod::InitSizeLists(const ShapeVector &keys_shape, const ShapeVector &values_shape) { in InitSizeLists()
|
| D | map_tensor_get_data_gpu_kernel.cc | 83 const auto &values_shape = outputs.at(kIndex1)->GetShapeVector(); in Resize() local
110 …etDataGpuKernelMod::InitSizeLists(const ShapeVector &keys_shape, const ShapeVector &values_shape) { in InitSizeLists()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ge/ |
| D | run_sparse_tensor.py | 41 def coo_data_generate(values_shape): argument
82 def csr_data_generate(values_shape): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_cummax.cc | 35 ShapeVector values_shape; member
48 auto values_shape = std::make_shared<abstract::Shape>(param.values_shape); in TEST_P() local
|
| D | test_ops_cummin.cc | 35 ShapeVector values_shape; member
48 auto values_shape = std::make_shared<abstract::Shape>(param.values_shape); in TEST_P() local
|
| D | test_ops_searchsorted.cc | 34 std::vector<int64_t> values_shape; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | sparse_count_sparse_output.cc | 49 auto values_shape = input_args[kInputIndex1]->GetShape()->GetShapeVector(); in SparseCountSparseOutputInferShape() local
98 …auto values_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->GetS… in SparseCountSparseOutputFrontendInferShape() local
|
| D | lower_bound.cc | 45 …auto values_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->GetShape())[kSh… in LowerBoundInferShape() local
|
| D | upper_bound.cc | 45 …auto values_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->GetShape())[kSh… in UpperBoundInferShape() local
|
| D | sparse_matrix_softmax.cc | 58 …auto values_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex4]->GetS… in SparseMatrixSoftmaxInferShape() local
|
| D | sparse_dense_cwise_arithmetic.cc | 65 …auto values_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->GetS… in SparseDenseCwiseArithmeticInferShape() local
|
| D | sparse_tensor_dense_mat_mul.cc | 107 … const ShapeVector &values_shape, const ShapeVector &shape_shape, in SparseTensorDenseMatmulCheckShape()
159 …auto values_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->GetShape())[kSh… in SparseTensorDenseMatmulInferShape() local
|
| D | sparse_to_dense.cc | 43 …auto values_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->GetS… in SparseToDenseInferShape() local
|
| D | sparse_reorder.cc | 42 auto values_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(values_shape_ptr)[kShape]; in SparseReorderInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | searchsorted.cc | 76 auto values_shape = values_shape_ptr->GetShapeVector(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/ |
| D | sparse_softmax_gpu_kernel.cc | 51 …std::vector<int64_t> values_shape = std::vector<int64_t>(inputs.at(kIndex1)->GetDeviceShapeVector(… in Resize() local
|
| D | sparse_reorder_gpu_kernel.cc | 66 auto values_shape = inputs.at(kIndex1)->GetShapeVector(); in Resize() local
|
| D | sparse_dense_cwise_operation_gpu_kernel.cc | 103 …std::vector<int64_t> values_shape = std::vector<int64_t>(inputs.at(kIndex1)->GetDeviceShapeVector(… in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | sparse_to_dense_cpu_kernal.cc | 50 auto values_shape = inputs.at(kIndex1)->GetShapeVector(); in Resize() local
|
| D | set_size_cpu_kernel.cc | 39 auto values_shape = inputs[kIndex1]->GetShapeVector(); in Resize() local
|
| D | sparse_reorder_cpu_kernal.cc | 51 auto values_shape = inputs.at(kIndex1)->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | ctcgreedydecoder_gpu_kernel.cc | 190 std::vector<int64_t> values_shape = outputs[kIndex1]->GetShapeVector(); in UpdateOutputShapeAndSize() local
|