Searched defs:src_dims (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/ |
| D | gru_cpu_kernel.cc | 106 dim src_dims = {seq_len_, batch_size_, input_size_}; in Resize() local
|
| D | matmul_cpu_kernel_func.cc | 95 dims src_dims, weights_dims, bias_dims, dst_dims, a_strides, b_strides, o_strides, bias_strides; in Resize() local
|
| D | lstm_cpu_kernel.cc | 110 dim src_dims = {seq_len_, batch_size_, input_size_}; in Resize() local
|
| D | gru_grad_cpu_kernel.cc | 121 dim src_dims = {seq_len_, batch_size_, input_size_}; in InitDnnl() local
|
| D | lstm_grad_cpu_kernel.cc | 124 dim src_dims = {seq_len_, batch_size_, input_size_}; in InitDnnl() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/ |
| D | node_util.cc | 53 … NodeUtils::ConvertDims(mindspore::schema::Format src_format, const std::vector<int32_t> &src_dims, in ConvertDims()
|