Home
last modified time | relevance | path

Searched defs:new_concat (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dconcat_opt.cpp121 auto new_concat = in handleCat() local
166 auto new_concat = in handleCat() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dvariadic_op_splitter.cc68 auto new_concat = comp->AddInstruction(concat->CloneWithNewOperands( in SplitConcatenate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_simplifier.cc46 auto new_concat = parent->AddInstruction(HloInstruction::CreateConcatenate( in ConcatForwarding() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Ddevice_transform_patterns.cc583 auto new_concat = rewriter.create<TFL::ConcatenationOp>( in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc916 auto new_concat = builder.create<TFL::ConcatenationOp>( in fold() local