Searched defs:transpose_op (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/ |
| D | insert_format_transform_op.cc | 63 auto transpose_op = graph->NewCNode(transpose_input); in InsertTransposeOp() local
107 auto transpose_op = InsertTransposeOp(graph, node, used_node, used_node_index, transpose_perm); in ProcessForTupleItem() local
130 auto transpose_op = InsertTransposeOp(graph, input_node, node, i, transpose_perm); in InsertTransformOpForInput() local
166 … auto transpose_op = InsertTransposeOp(graph, node, used_node, used_node_index, transpose_perm); in InsertTransformOpForOutput() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | insert_format_transform_op.cc | 99 auto transpose_op = graph->NewCNode(transpose_input); in InsertTransposeOp() local
185 auto transpose_op = InsertTransposeOp(graph, node, used_node, used_node_index, transpose_perm); in ProcessForTupleItem() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/pass/ |
| D | npu_pass_utils.cc | 148 auto transpose_op = static_cast<TransposeNPUOp *>(op); in IsNhwc2Nchw() local
164 auto transpose_op = static_cast<TransposeNPUOp *>(op); in IsNchw2Nhwc() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | shuffle_tensorrt.cc | 211 auto transpose_op = this->op_primitive_->value_as_Transpose(); in AddTransposeOp() local
|
| /third_party/ffmpeg/libavfilter/dnn/ |
| D | dnn_backend_tf.c | 317 static DNNReturnType add_conv_layer(TFModel *tf_model, TF_Operation *transpose_op, TF_Operation **c… in add_conv_layer()
573 TF_Operation *transpose_op; in load_native_model() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/ |
| D | npu_graph.cc | 288 auto transpose_op = static_cast<TransposeNPUOp *>(op); in CreateNPUTransposeKernel() local
|