| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| remove_trivial_concatenation.cc | 33 auto* concat_op = concat_it->get(); local
34 if (concat_op->type != OperatorType::kConcatenation) {
37 if (concat_op->inputs.size() != 1) {
|
| remove_trivial_concatenation_input.cc | 40 auto* concat_op = concat_it->get(); local
41 if (concat_op->type != OperatorType::kConcatenation) {
46 for (const string& input : concat_op->inputs) {
68 concat_op->inputs = nontrivial_inputs;
|
| convert_trivial_tile_to_concat.cc | 72 auto* concat_op = new ConcatenationOperator; local
77 concat_op->inputs.push_back(tile_op->inputs[0]);
79 concat_op->axis = concat_axis;
80 concat_op->outputs = tile_op->outputs;
89 const auto concat_it = model->operators.emplace(tile_it, concat_op);
|
| resolve_constant_concatenation.cc | 147 const auto* concat_op = local
150 for (const string& input_name : concat_op->inputs) {
166 const int concatenation_axis = concat_op->axis;
168 CHECK_EQ(concat_op->outputs.size(), 1);
169 string concatenated_array_name = concat_op->outputs[0];
172 for (const string& input_name : concat_op->inputs) {
177 absl::StrJoin(concat_op->inputs, ", "), concatenated_array_name);
210 for (const string& input_name : concat_op->inputs) {
|
| /external/tensorflow/tensorflow/python/ops/ |
| while_v2_indexed_slices_rewriter.py | 224 concat_op = old_output.inputs[0].op
225 assert concat_op.type == "ConcatV2"
227 old_concat_args = concat_op.inputs[:-1]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_concat_op.cc | 417 auto concat_op = concat(concat_pd, inputs, dst.GetOpMem()); local
419 net.push_back(concat_op);
|
| /external/tensorflow/tensorflow/lite/toco/ |
| export_tensorflow.cc | 1467 tensorflow::NodeDef* concat_op = tensorflow_graph->add_node(); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier.cc | [all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fold_old_batch_norms_test.cc | 396 Output concat_op = local
397 Concat(root.WithOpName("concat_op"), {conv0_op, conv1_op}, shape_op);
425 AddNodeInput("concat_op", &batch_norm_node);
|
| quantize_nodes_test.cc | 385 Output concat_op = local
386 Concat(root.WithOpName("concat_op"), {a_op, b_op}, shape_op);
391 TestQuantizedVersusFloatGraph(float_graph_def, {}, {"concat_op"});
[all...] |
