Searched defs:final_shape (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | get_tuple_index_info.cc | 157 …st std::vector<int64_t> &tuple_index_types, ShapeVector *broadcast_shape, ShapeVector *final_shape, in ConstGetTupleIndexInfo()
292 ShapeVector final_shape; in GetTupleIndexInfoInferInner() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | get_tuple_index_info_cpu_kernel.cc | 71 ShapeVector final_shape; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | sparse_fill_empty_rows.cu | 211 … size_t *final_shape, int64_t *origin_index, int64_t *sorted_key, in SparseFillEmptyRows()
243 … size_t *final_shape, int64_t *origin_index, int64_t *sorted_key, in SparseFillEmptyRows()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_index_py.cc | 357 ShapeVector final_shape = slice_shapes; in GenerateIndexInfoFromTupleOfMixedTensors() local
390 TensorIndex TensorIndex::SliceToArray(const TensorPtr &index, const ShapeVector &final_shape, size_… in SliceToArray()
406 …ex TensorIndex::BroadCastTensor(const ShapeVector &broadcast_shape, const ShapeVector &final_shape, in BroadCastTensor()
640 const std::vector<int64_t> &final_shape, in GenerateIndices()
735 ShapeVector final_shape = std::get<final_shape_index>(index_info); in TensorGetitemByTuple() local
1086 ShapeVector final_shape = std::get<final_shape_index>(index_info); in GenerateIndicesFromTuple() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/multitype_ops/ |
| D | _compile_utils.py | 400 def _transform_indexing_tensor(broadcast_shape, final_shape, new_shape, item): argument
895 def _tensor_index_transfer(index, broadcast_shape, final_shape, new_shape): argument
1436 def convert_slice_to_tensor(index, final_shape, slice_cnt, broadcast_shape, slice_shapes, fancy_pos… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_concat.cc | 178 std::vector<int64_t> final_shape(tensors[0]->shape_.begin(), tensors[0]->shape_.end()); in Concat() local
|