/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
H A D | remove_trivial_concatenation.cc | 33 auto* concat_op = concat_it->get(); in Run() local
|
H A D | remove_trivial_concatenation_input.cc | 40 auto* concat_op = concat_it->get(); in Run() local
|
H A D | convert_trivial_tile_to_concat.cc | 72 auto* concat_op = new ConcatenationOperator; in Run() local
|
H A D | resolve_constant_concatenation.cc | 148 const auto* concat_op = in Run() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
H A D | device_transform_patterns.cc | 249 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 D | dot_merger.cc | 175 HloInstruction* concat_op = in TryMergeSameOperand() local
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
H A D | fold_old_batch_norms_test.cc | 397 Output concat_op = in TestFoldFusedBatchNormsWithConcat() local
|
H A D | quantize_nodes_test.cc | 385 Output concat_op = in TestQuantizeConcat() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
H A D | operation_selector.cc | 402 auto& concat_op = gpu_subgraph->operations[g]; in GPUOperationFromNodePart0() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
H A D | xla_sharding_util.cc | 451 mlir::TF::ConcatOp concat_op = in HandleTileShardedOutputs() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
H A D | mkl_concat_op.cc | 755 auto concat_op = concat(concat_pd); in Compute() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
H A D | lower_static_tensor_list.cc | 835 auto concat_op = rewriter->create<TF::ConcatOp>( in CreateCondTrueBranch() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
H A D | lower_tf.cc | 1627 auto concat_op = rewriter.create<ConcatV2Op>( in matchAndRewrite() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
H A D | generic_layout_optimizer_transposer_test.cc | 2790 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 D | export_tensorflow.cc | 1502 tensorflow::NodeDef* concat_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
H A D | pass.cc | 2860 Operation *concat_op = rewriter.create(state); in matchAndRewrite() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
H A D | legalize_common.cc | 661 auto concat_op = CreateOpAndInfer<tosa::ConcatOp>( in convertConcatV2Op() local
|