Home
last modified time | relevance | path

Searched refs:concat_inputs (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_lstm.cc239 Operator* concat_inputs; in Run() local
241 OperatorType::kConcatenation, &concat_inputs, in Run()
256 lstm_cell_op->inputs[LstmCellOperator::DATA_INPUT] = concat_inputs->inputs[0]; in Run()
258 concat_inputs->inputs[1]; in Run()
281 model->GetArray(concat_inputs->outputs[0]).data_type; in Run()
304 DeleteOpAndArrays(model, concat_inputs); in Run()
/external/tensorflow/tensorflow/compiler/tests/
Dconcat_ops_test.py104 concat_inputs = [math_ops.cast(p_i, dtype) for p_i in p]
106 concat_inputs = p
108 c = array_ops.concat(concat_inputs, concat_dim)
317 concat_inputs = p
318 c = array_ops.concat(concat_inputs, concat_dim)
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner_test.cc1066 OutputList concat_inputs{ in TEST_F() local
1073 auto concat = ops::Concat(root, concat_inputs, concat_dim); in TEST_F()
1084 for (const auto& o : concat_inputs) { in TEST_F()
1108 OutputList concat_inputs{ in TEST_F() local
1115 auto concat = ops::Concat(root, concat_inputs, concat_dim); in TEST_F()
1127 for (const auto& o : concat_inputs) { in TEST_F()
1150 OutputList concat_inputs{ in TEST_F() local
1157 auto concat = ops::Concat(root, concat_inputs, concat_dim); in TEST_F()
1168 for (const auto& o : concat_inputs) { in TEST_F()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconcat_op_test.py136 concat_inputs = [math_ops.cast(p_i, dtype) for p_i in p]
138 concat_inputs = p
139 c = array_ops.concat(concat_inputs, concat_dim)
545 concat_inputs = p
546 c = array_ops.concat(concat_inputs, concat_dim)
Drnn_cell_test.py1156 concat_inputs = array_ops.placeholder(
1159 concat_inputs = constant_op.constant(input_values)
1160 inputs = array_ops.unstack(concat_inputs)
1180 outputs_static + [state_static], [concat_inputs])
1183 gradients_impl.gradients(y, [concat_inputs])
1197 feeds = {concat_inputs: input_values}
1216 concat_inputs = array_ops.placeholder(
1219 concat_inputs = constant_op.constant(input_values)
1236 inputs=concat_inputs,
1246 split_outputs_dynamic + [state_dynamic], [concat_inputs])
[all …]
/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_testlib.py575 concat_inputs = array_ops.placeholder(
580 inputs=concat_inputs,
601 sess.run(train_op, feed_dict={concat_inputs: input_values},
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2695 std::vector<nvinfer1::ITensor const*> concat_inputs; in DynamicReshape() local
2705 concat_inputs.push_back(tensor); in DynamicReshape()
2711 concat_inputs.push_back(slice_layer->getOutput(0)); in DynamicReshape()
2716 const_cast<nvinfer1::ITensor* const*>(concat_inputs.data()), in DynamicReshape()
2717 concat_inputs.size()); in DynamicReshape()