Searched defs:transpose_op (Results 1 – 13 of 13) sorted by relevance
108 auto* transpose_op = new TransposeOperator; in Run() local136 auto* transpose_op = FindTransposeOpWithInput(*model, input_rhs); in Run() local
63 auto* transpose_op = new TransposeOperator; in CreateTransposeFromReorderAxes() local139 auto* transpose_op = CreateTransposeFromReorderAxes( in Run() local
119 auto* transpose_op = new TransposeOperator; in TransposeInput() local154 TransposeOperator* transpose_op = TransposeInput(input_lhs, model); in Run() local162 TransposeOperator* transpose_op = TransposeInput(input_rhs, model); in Run() local
59 TransposeOperator* transpose_op = in Run() local
135 TransposeOperator* transpose_op = ConvertOperator<TransposeOperator*>( in Run() local
110 TransposeOperator* transpose_op = ConvertOperator<TransposeOperator*>( in Run() local
72 auto* transpose_op = new toco::TransposeOperator; in CreateGraph() local130 auto* transpose_op = new toco::TransposeOperator; in TEST_F() local
134 auto transpose_op = fused_lstm_func_.getBody().front().begin(); in TEST_F() local
300 auto transpose_op = GetTransposeOpForConv2DFilter(&builder, reshape_op); in HandleConv2DFilter() local390 auto transpose_op = GetTransposeOpForConv2DFilter(&builder, first_reshape_op); in HandleConv2DBackPropFilter() local
241 auto transpose_op = *transpose_ops.begin(); in MoveTransposeBefore() local
220 auto& transpose_op = gpu_subgraph->operations[0]; in GPUOperationFromNode() local
635 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local1295 tensorflow::NodeDef* transpose_op = tensorflow_graph->add_node(); in ConvertTransposeOperator() local
2403 const NodeDef* transpose_op = node->op() == "Conj" ? input : node; in TrySimplify() local