Home
last modified time | relevance | path

Searched defs:input_indices_shape (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/online_fusion/
Dcast_gather_reduce_fp32.cc44 auto input_indices_shape = in_tensors_.at(1)->shape(); in DoCastGatherReduceFusion() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_gather.cc25 ShapeVector input_indices_shape; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dembedding_lookup_gpu_kernel.cc196 std::vector<int64_t> input_indices_shape = inputs[kIndex1]->GetShapeVector(); in Resize() local
Dcoalesce_gpu_kernel.cc100 auto input_indices_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/
Dsparse_slice_gpu_kernel.cc46 auto input_indices_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
Dsparse_split_gpu_kernel.cc61 auto input_indices_shape = inputs[kIndex1]->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse_grad/
Dsparse_slice_grad_gpu_kernel.cc73 auto input_indices_shape = inputs.at(kIndex1)->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_dense_embedding_operation.cc318 ShapeVector input_indices_shape = {SizeToLong(indices_num)}; in LookupDeviceCache() local
379 const ShapeVector input_indices_shape = {SizeToLong(indices_num)}; in UpdateDeviceCache() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsparse_slice_cpu_kernel.cc62 const auto input_indices_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
Dembedding_look_up_cpu_kernel.cc151 std::vector<int64_t> input_indices_shape = inputs[kIndex1]->GetShapeVector(); in Resize() local
Dsspaddmm_cpu_kernel.cc67 auto input_indices_shape = inputs.at(kIndex0)->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsspaddmm.cc100 auto input_indices_shape = input_indices_tensor->GetTensorShape(); in CheckInputDims() local