| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | remove_trivial_concatenation_input.cc | 40 auto* concat_op = concat_it->get(); in Run() local
41 if (concat_op->type != OperatorType::kConcatenation) { in Run()
46 for (const std::string& input : concat_op->inputs) { in Run()
62 concat_op->inputs = nontrivial_inputs; in Run()
64 DeleteArrayIfUnusedOutsideOfOp(input, concat_op, model); in Run()
|
| D | convert_trivial_tile_to_concat.cc | 72 auto* concat_op = new ConcatenationOperator; in Run() local
77 concat_op->inputs.push_back(tile_op->inputs[0]); in Run()
79 concat_op->axis = concat_axis; in Run()
80 concat_op->outputs = tile_op->outputs; in Run()
89 model->operators.emplace(tile_it, concat_op); in Run()
|
| D | resolve_constant_concatenation.cc | 147 const auto* concat_op = in Run() local
150 for (const std::string& input_name : concat_op->inputs) { in Run()
166 const int concatenation_axis = concat_op->axis; in Run()
168 CHECK_EQ(concat_op->outputs.size(), 1); in Run()
169 std::string concatenated_array_name = concat_op->outputs[0]; in Run()
172 for (const std::string& input_name : concat_op->inputs) { in Run()
177 absl::StrJoin(concat_op->inputs, ", "), concatenated_array_name); in Run()
209 DeleteOpAndArrays(model, concat_op); in Run()
|
| D | remove_trivial_concatenation.cc | 33 auto* concat_op = concat_it->get(); in Run() local
34 if (concat_op->type != OperatorType::kConcatenation) { in Run()
37 if (concat_op->inputs.size() != 1) { in Run()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2_indexed_slices_rewriter.py | 233 concat_op = old_output.inputs[0].op
234 assert concat_op.type == "ConcatV2"
236 old_concat_args = concat_op.inputs[:-1]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 106 mlir::TF::ConcatOp* concat_op) { in CreateConcatOp() argument
136 *concat_op = builder->create<mlir::TF::ConcatOp>( in CreateConcatOp()
437 mlir::TF::ConcatOp concat_op; in HandleTileShardedOutputs() local
443 builder, &concat_op); in HandleTileShardedOutputs()
446 new_outputs.emplace_back(concat_op.getResult()); in HandleTileShardedOutputs()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1497 tensorflow::NodeDef* concat_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
1498 concat_op->set_op("ConcatV2"); in ConvertLstmCellOperator()
1499 concat_op->set_name(concat_output); in ConvertLstmCellOperator()
1500 *concat_op->add_input() = src_op.inputs[LstmCellOperator::DATA_INPUT]; in ConvertLstmCellOperator()
1501 *concat_op->add_input() = src_op.inputs[LstmCellOperator::PREV_ACTIV_INPUT]; in ConvertLstmCellOperator()
1502 *concat_op->add_input() = axis_output; in ConvertLstmCellOperator()
1503 (*concat_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertLstmCellOperator()
1504 (*concat_op->mutable_attr())["Tidx"].set_type(DT_INT32); in ConvertLstmCellOperator()
1505 (*concat_op->mutable_attr())["N"].set_i(2); // Number of inputs in ConvertLstmCellOperator()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 742 auto concat_op = concat(concat_pd); in Compute() local
748 concat_op.execute(*fwd_cpu_stream, net_args); in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 673 auto concat_op = rewriter->create<TF::ConcatOp>( in CreateCondTrueBranch() local
676 rewriter->create<ReturnOp>(loc, ArrayRef<Value>({concat_op})); in CreateCondTrueBranch()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 27 "concat_op.cc",
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 687 auto concat_op = rewriter.create<tosa::ConcatOp>( in convertConcatV2Op() local
722 concat_op = rewriter.create<tosa::ConcatOp>( in convertConcatV2Op()
723 op->getLoc(), concat_type, concat_op.getResult(), rhs_val, in convertConcatV2Op()
727 return concat_op.getResult(); in convertConcatV2Op()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 342 …%concat_op = tosa.CONCAT(%values:0, %values:1) {axis=tosa_axis} : (tensor<%values:0.type>, tensor<…
346 …%concat_op = tosa.CONCAT(%concat_op, %values:i) {axis=tosa_axis} : (tensor<%concat_op.type>, tenso…
349 return %concat_op
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 397 Output concat_op = in TestFoldFusedBatchNormsWithConcat() local
|
| D | quantize_nodes_test.cc | 385 Output concat_op = in TestQuantizeConcat() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 2790 Output concat_op; in TEST_F() local
2793 {input_1, input_2, conv2d}, axis, &concat_op)); in TEST_F()
2794 auto z = ops::Identity(scope.WithOpName("z"), concat_op); in TEST_F()
2868 auto concat_op = in TEST_F() local
2871 auto z = ops::Identity(scope.WithOpName("z"), concat_op); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 960 ":concat_op",
1017 name = "concat_op",
1018 prefix = "concat_op",
1562 ":concat_op",
5786 "concat_op.cc",
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 1908 for (HloInstruction* concat_op : lhs->operands()) { in OptimizeDotOfConcatHelper()
1909 int64 sub_k = concat_op->shape().dimensions(lhs_contracting_dim); in OptimizeDotOfConcatHelper()
1935 new_dot_rhs = concat_op; in OptimizeDotOfConcatHelper()
1937 new_dot_lhs = concat_op; in OptimizeDotOfConcatHelper()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 2084 "//tensorflow/core/kernels:concat_op",
|