Searched refs:concat_op (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_concatenation.cc | 147 const auto* concat_op = in Run() local
150 for (const 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 string concatenated_array_name = concat_op->outputs[0]; in Run()
172 for (const string& input_name : concat_op->inputs) { in Run()
177 absl::StrJoin(concat_op->inputs, ", "), concatenated_array_name); in Run()
210 for (const string& input_name : concat_op->inputs) { 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 const auto concat_it = model->operators.emplace(tile_it, concat_op); in Run()
|
| 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 string& input : concat_op->inputs) { in Run()
68 concat_op->inputs = nontrivial_inputs; 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 | 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/ |
| D | mkl_concat_op.cc | 417 auto concat_op = concat(concat_pd, inputs, dst.GetOpMem()); in Compute() local
419 net.push_back(concat_op); in Compute()
|
| D | BUILD | 874 ":concat_op",
935 name = "concat_op",
936 prefix = "concat_op",
1345 ":concat_op",
5480 "concat_op.cc",
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1467 tensorflow::NodeDef* concat_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
1468 concat_op->set_op("ConcatV2"); in ConvertLstmCellOperator()
1469 concat_op->set_name(concat_output); in ConvertLstmCellOperator()
1470 *concat_op->add_input() = src_op.inputs[LstmCellOperator::DATA_INPUT]; in ConvertLstmCellOperator()
1471 *concat_op->add_input() = src_op.inputs[LstmCellOperator::PREV_ACTIV_INPUT]; in ConvertLstmCellOperator()
1472 *concat_op->add_input() = axis_output; in ConvertLstmCellOperator()
1473 (*concat_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertLstmCellOperator()
1474 (*concat_op->mutable_attr())["Tidx"].set_type(DT_INT32); in ConvertLstmCellOperator()
1475 (*concat_op->mutable_attr())["N"].set_i(2); // Number of inputs in ConvertLstmCellOperator()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 26 "concat_op.cc",
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 396 Output concat_op = in TestFoldFusedBatchNormsWithConcat() local
|
| D | quantize_nodes_test.cc | 385 Output concat_op = in TestQuantizeConcat() local
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 39 tensorflow/core/kernels/concat_op.cc
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 1455 for (HloInstruction* concat_op : lhs->operands()) { in OptimizeDotOfConcatHelper()
1456 int64 sub_k = concat_op->shape().dimensions(lhs_contracting_dim); in OptimizeDotOfConcatHelper()
1481 new_dot_rhs = concat_op; in OptimizeDotOfConcatHelper()
1483 new_dot_lhs = concat_op; in OptimizeDotOfConcatHelper()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 4401 "//tensorflow/core/kernels:concat_op",
|