Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dremove_trivial_concatenation.cc33 auto* concat_op = concat_it->get(); in Run() local
Dremove_trivial_concatenation_input.cc40 auto* concat_op = concat_it->get(); in Run() local
Dconvert_trivial_tile_to_concat.cc72 auto* concat_op = new ConcatenationOperator; in Run() local
Dresolve_constant_concatenation.cc147 const auto* concat_op = in Run() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc106 mlir::TF::ConcatOp* concat_op) { in CreateConcatOp()
437 mlir::TF::ConcatOp concat_op; in HandleTileShardedOutputs() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_old_batch_norms_test.cc397 Output concat_op = in TestFoldFusedBatchNormsWithConcat() local
Dquantize_nodes_test.cc385 Output concat_op = in TestQuantizeConcat() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_concat_op.cc742 auto concat_op = concat(concat_pd); in Compute() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc673 auto concat_op = rewriter->create<TF::ConcatOp>( in CreateCondTrueBranch() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc2790 Output concat_op; in TEST_F() local
2868 auto concat_op = in TEST_F() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1497 tensorflow::NodeDef* concat_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc687 auto concat_op = rewriter.create<tosa::ConcatOp>( in convertConcatV2Op() local