Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_reorder_axes.cc63 auto* transpose_op = new TransposeOperator; in CreateTransposeFromReorderAxes() local
139 auto* transpose_op = CreateTransposeFromReorderAxes( in Run() local
Dunroll_batch_matmul.cc190 auto* transpose_op = new TransposeOperator; in TransposeInput() local
235 TransposeOperator* transpose_op = TransposeInput(input_lhs, model); in Run() local
243 TransposeOperator* transpose_op = TransposeInput(input_rhs, model); in Run() local
Dresolve_tensorflow_matmul.cc92 auto* transpose_op = FindTransposeOpWithInput(*model, input_rhs); in Run() local
Dconvert_trivial_transpose_to_reshape.cc59 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/
Dexport_tensorflow.cc630 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local
1282 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); in ConvertTransposeOperator() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc2290 const NodeDef* transpose_op = node->op() == "Conj" ? input : node; in TrySimplify() local