Searched defs:concat_op (Results 1 – 15 of 15) sorted by relevance
33 auto* concat_op = concat_it->get(); in Run() local
40 auto* concat_op = concat_it->get(); in Run() local
72 auto* concat_op = new ConcatenationOperator; in Run() local
147 const auto* concat_op = in Run() local
245 auto concat_op = rewriter.create<TFL::ConcatenationOp>( in matchAndRewrite() local525 LogicalResult PadConcat::matchAndRewrite(TFL::ConcatenationOp concat_op, in matchAndRewrite()
277 auto& concat_op = gpu_subgraph->operations[g]; in GPUOperationFromNode() local
441 mlir::TF::ConcatOp concat_op = in HandleTileShardedOutputs() local
397 Output concat_op = in TestFoldFusedBatchNormsWithConcat() local
385 Output concat_op = in TestQuantizeConcat() local
745 auto concat_op = concat(concat_pd); in Compute() local
752 auto concat_op = rewriter->create<TF::ConcatOp>( in CreateCondTrueBranch() local
1635 auto concat_op = rewriter.create<ConcatV2Op>( in matchAndRewrite() local
2790 Output concat_op; in TEST_F() local2868 auto concat_op = in TEST_F() local
1497 tensorflow::NodeDef* concat_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
631 auto concat_op = rewriter.create<tosa::ConcatOp>( in convertConcatV2Op() local