Home
last modified time | relevance | path

Searched defs:transpose_op (Results 1 – 15 of 15) sorted by relevance

/external/XNNPACK/test/
Dtranspose-operator-tester.h73 xnn_operator_t transpose_op = nullptr; in TestX8() local
142 xnn_operator_t transpose_op = nullptr; in TestX16() local
211 xnn_operator_t transpose_op = nullptr; in TestX32() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_tensorflow_matmul.cc109 auto* transpose_op = new TransposeOperator; in Run() local
137 auto* transpose_op = FindTransposeOpWithInput(*model, input_rhs); in Run() local
Dconvert_reorder_axes.cc63 auto* transpose_op = new TransposeOperator; in CreateTransposeFromReorderAxes() local
139 auto* transpose_op = CreateTransposeFromReorderAxes( in Run() local
Dunroll_batch_matmul.cc121 auto* transpose_op = new TransposeOperator; in TransposeInput() local
156 TransposeOperator* transpose_op = TransposeInput(input_lhs, model); in Run() local
164 TransposeOperator* transpose_op = TransposeInput(input_rhs, model); in Run() local
Dconvert_trivial_transpose_to_reshape.cc60 TransposeOperator* transpose_op = in Run() local
Dmerge_reshape_into_preceding_transpose.cc135 TransposeOperator* transpose_op = ConvertOperator<TransposeOperator*>( in Run() local
Dreorder_reshape_transpose.cc110 TransposeOperator* transpose_op = ConvertOperator<TransposeOperator*>( in Run() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dremove_successive_transpose_test.cc73 auto* transpose_op = new toco::TransposeOperator; in CreateGraph() local
131 auto* transpose_op = new toco::TransposeOperator; in TEST_F() local
/external/XNNPACK/src/operators/
Dtranspose-nd.c39 xnn_operator_t transpose_op) in init_transpose_nd()
73 xnn_operator_t transpose_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in create_transpose_nd() local
104 xnn_operator_t transpose_op, in setup_transpose_nd()
352 xnn_operator_t transpose_op, in xnn_setup_transpose_nd_x32()
375 xnn_operator_t transpose_op, in xnn_setup_transpose_nd_x16()
398 xnn_operator_t transpose_op, in xnn_setup_transpose_nd_x8()
439 struct xnn_operator transpose_op; in run_transpose_nd() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils_test.cc134 auto transpose_op = fused_lstm_func_.getBody().front().begin(); in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc255 auto transpose_op = GetTransposeOpForConv2DFilter(&builder, reshape_op); in HandleConv2DFilter() local
345 auto transpose_op = GetTransposeOpForConv2DFilter(&builder, first_reshape_op); in HandleConv2DBackPropFilter() local
Dlayout_optimization.cc222 auto transpose_op = *transpose_ops.begin(); in MoveTransposeBefore() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc640 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local
1300 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); in ConvertTransposeOperator() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc2637 auto transpose_op = CreateOpAndInfer<tosa::TransposeOp>( in convertGenericReduceOp() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc2496 const NodeDef* transpose_op = node->op() == "Conj" ? input : node; in TrySimplify() local