Home
last modified time | relevance | path

Searched refs:new_concat (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dvariadic_op_splitter.cc68 auto new_concat = comp->AddInstruction(concat->CloneWithNewOperands( in SplitConcatenate() local
70 new_operands.push_back(new_concat); in SplitConcatenate()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc764 auto new_concat = builder.create<TFL::ConcatenationOp>( in fold() local
768 return new_concat.getResult(); in fold()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc2583 auto new_concat = rewriter.create<ConcatenateOp>( in matchAndRewrite() local
2593 slice.getLoc(), new_concat, in matchAndRewrite()