| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/ |
| D | scale_npu.cc | 168 std::vector<int> valid_shape(valid_dims, 1); in ConvertScaleToMul() local
241 std::vector<int> valid_shape; in Adopt4DScale() local
269 std::vector<int> valid_shape(actual_dim, 1); in ChangeDims() local
|
| D | flatten_npu.cc | 58 std::vector<int> valid_shape; in SetNPUInputs() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/tbe/ |
| D | tbe_fusion_common.json | 58 "valid_shape": [] array
93 "valid_shape": [] array
132 "valid_shape": [] array
163 "valid_shape": [] array
202 "valid_shape": [] array
241 "valid_shape": [] array
272 "valid_shape": [] array
|
| D | tbe_fusion_add_conv2d.json | 58 "valid_shape": [] array
92 "valid_shape": [] array
126 "valid_shape": [] array
183 "valid_shape": [] array
211 "valid_shape": [] array
250 "valid_shape": [] array
281 "valid_shape": [] array
396 "valid_shape": [] array
424 "valid_shape": [] array
473 "valid_shape": [] array
[all …]
|
| D | tbe_single_layer_norm_select.json | 1 …[2,3],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"D… array
|
| D | tbe_single_dynamic_rnn_build.json | 1 …4,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"D… array
|
| D | tbe_single_dynamic_rnn_select.json | 1 …6,64],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"D… array
|
| D | tbe_single_layer_norm_build.json | 1 …1,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"D… array
|
| D | tbe_single_common_select.json | 1 …,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"D… array
|
| D | tbe_single_common_build.json | 1 …4,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"D… array
|
| D | tbe_single_conv2d_backprop_filter_build.json | 1 …4,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"D… array
|
| D | tbe_single_conv2d_backprop_filter_select.json | 1 …,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"D… array
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/other/ |
| D | check_valid_gpu_kernel.cc | 55 auto valid_shape = outputs.at(kIndex0)->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | dynamic_gru_v2.cc | 97 std::vector<int64_t> valid_shape = {3 * hidden_size}; in DynamicGRUV2InferShape() local
|
| D | dynamic_gru_v2_grad.cc | 111 std::vector<int64_t> valid_shape = {num_step, batch_size, hidden_size}; in DynamicGRUV2GradCheckShapeValue() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | dynamic_rnn_grad.cc | 145 std::vector<int64_t> valid_shape{x_shape[kDynRNNGradIdx0], x_shape[kDynRNNGradIdx1], hidden_size}; in DynamicRNNGradInferShape() local
|