Home
last modified time | relevance | path

Searched defs:concat_op (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dremove_trivial_concatenation.cc33 auto* concat_op = concat_it->get(); in Run() local
H A Dremove_trivial_concatenation_input.cc40 auto* concat_op = concat_it->get(); in Run() local
H A Dconvert_trivial_tile_to_concat.cc72 auto* concat_op = new ConcatenationOperator; in Run() local
H A Dresolve_constant_concatenation.cc148 const auto* concat_op = in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Ddevice_transform_patterns.cc249 auto concat_op = rewriter.create<TFL::ConcatenationOp>( in matchAndRewrite() local
529 LogicalResult PadConcat::matchAndRewrite(TFL::ConcatenationOp concat_op, in matchAndRewrite()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddot_merger.cc175 HloInstruction* concat_op = in TryMergeSameOperand() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_old_batch_norms_test.cc397 Output concat_op = in TestFoldFusedBatchNormsWithConcat() local
H A Dquantize_nodes_test.cc385 Output concat_op = in TestQuantizeConcat() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Doperation_selector.cc402 auto& concat_op = gpu_subgraph->operations[g]; in GPUOperationFromNodePart0() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dxla_sharding_util.cc451 mlir::TF::ConcatOp concat_op = in HandleTileShardedOutputs() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_concat_op.cc755 auto concat_op = concat(concat_pd); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dlower_static_tensor_list.cc835 auto concat_op = rewriter->create<TF::ConcatOp>( in CreateCondTrueBranch() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlower_tf.cc1627 auto concat_op = rewriter.create<ConcatV2Op>( in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer_test.cc2790 Output concat_op; in TEST_F() local
2868 auto concat_op = in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc1502 tensorflow::NodeDef* concat_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc2860 Operation *concat_op = rewriter.create(state); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_common.cc661 auto concat_op = CreateOpAndInfer<tosa::ConcatOp>( in convertConcatV2Op() local