Home
last modified time | relevance | path

Searched defs:split_index (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/tbe/
Dtbe_fusion_common.json56 "split_index": 0, number
91 "split_index": 0, number
130 "split_index": 0, number
161 "split_index": 0, number
200 "split_index": 0, number
239 "split_index": 0, number
270 "split_index": 0, number
Dtbe_fusion_add_conv2d.json56 "split_index": 0, number
90 "split_index": 0, number
124 "split_index": 0, number
181 "split_index": 0, number
209 "split_index": 0, number
248 "split_index": 0, number
279 "split_index": 0, number
394 "split_index": 0, number
422 "split_index": 0, number
471 "split_index": 0, number
[all …]
Dtbe_single_layer_norm_select.json1 …ge":[[1,1],[1,1],[3,3],[1,1],[16,16]],"shape":[2,3],"slice_offset":[],"split_index":0,"total_shape… number
Dtbe_single_dynamic_rnn_build.json1 …,1],[16,16],[64,64],[16,16]],"shape":[1,1,16,64,16],"slice_offset":[],"split_index":0,"total_shape… number
Dtbe_single_dynamic_rnn_select.json1 …1],[1,1],[16,16],[64,64],[16,16]],"shape":[2,16,64],"slice_offset":[],"split_index":0,"total_shape… number
Dtbe_single_layer_norm_build.json1 …,1],[1,1],[3,3],[1,1],[16,16]],"shape":[1,1,3,1,16],"slice_offset":[],"split_index":0,"total_shape… number
Dtbe_single_common_select.json1 …[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape… number
Dtbe_single_common_build.json1 …24,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape… number
Dtbe_single_conv2d_backprop_filter_build.json1 …24,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape… number
Dtbe_single_conv2d_backprop_filter_select.json1 …[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape… number
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/_custom_op/
Dtranspose02314_impl.py314 def _inner_compute(split_index): argument
336 def _inner_compute(split_index, block_idx): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dspeech_commands_op.cc177 int32_t split_index = 0; in GetFileInfo() local
/third_party/rust/rust/src/tools/miri/src/
Drange_map.rs111 fn split_index(&mut self, index: usize, split_offset: u64) -> bool in split_index() method
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsession_basic.cc1412 void CheckSplitIndexValid(const vector<uint32_t> &split_index) { in CheckSplitIndexValid()
1422 void PreProcessOnSplitIndex(const KernelGraphPtr &graph, vector<uint32_t> *split_index) { in PreProcessOnSplitIndex()