| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| remove_trivial_concatenation.cc | 30 auto* concat_op = concat_it->get(); local
31 if (concat_op->type != OperatorType::kConcatenation) {
34 if (concat_op->inputs.size() != 1) {
|
| remove_trivial_concatenation_input.cc | 37 auto* concat_op = concat_it->get(); local
38 if (concat_op->type != OperatorType::kConcatenation) {
43 for (const string& input : concat_op->inputs) {
65 concat_op->inputs = nontrivial_inputs;
|
| resolve_constant_concatenation.cc | 139 const auto* concat_op = local
142 for (const string& input_name : concat_op->inputs) {
154 const int concatenation_axis = concat_op->axis;
156 CHECK_EQ(concat_op->outputs.size(), 1);
157 string concatenated_array_name = concat_op->outputs[0];
160 for (const string& input_name : concat_op->inputs) {
191 for (const string& input_name : concat_op->inputs) {
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| export_tensorflow.cc | 1226 auto* concat_op = tensorflow_graph->add_node(); local
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fold_old_batch_norms_test.cc | 240 Output concat_op = local
241 Concat(root.WithOpName("concat_op"), {conv0_op, conv1_op}, shape_op);
269 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...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_concat_op.cc | 770 auto concat_op = concat(concat_pd, inputs, dst.GetOpMem()); local
772 net.push_back(concat_op);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier.cc | [all...] |
