Searched refs:current_input (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | strip_unused_nodes.cc | 139 for (const string& current_input : current_inputs) { in StripUnusedNodes() local
140 required_nodes.insert(current_input); in StripUnusedNodes()
141 if (input_nodes.count(current_input)) { in StripUnusedNodes()
144 if (!node_lookup.count(current_input)) { in StripUnusedNodes()
145 return errors::InvalidArgument("Input node ", current_input, in StripUnusedNodes()
148 const NodeDef* current_node = node_lookup[current_input]; in StripUnusedNodes()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | group_bidirectional_sequence_ops.cc | 457 std::string current_input = first_fw_sequence_input->outputs[0]; in GroupDynamicSequenceOps() local
459 operator_type, current_input, bidirectional_sequence_ops, &op_it, model); in GroupDynamicSequenceOps()
528 std::string current_input = first_fw_lstm_input->outputs[0]; in Run() local
530 OperatorType::kBidirectionalSequenceLstm, current_input, in Run()
604 std::string current_input = first_fw_rnn_input->outputs[0]; in Run() local
606 OperatorType::kBidirectionalSequenceRnn, current_input, in Run()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | functional_ops_test.py | 351 def fn(_, current_input): argument
352 return current_input
362 def fn(_, current_input): argument
363 return current_input
384 def fn(_, current_input): argument
385 return current_input
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 1165 def body(time, elements_finished, current_input, emit_ta, state, argument
1180 (next_output, cell_state) = cell(current_input, state)
1191 nest.assert_same_structure(current_input, next_input)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 4544 current_input = tuple(ta.read(time) for ta in input_ta)
4546 current_input = nest.pack_sequence_as(inputs, current_input)
4548 output, new_states = step_function(current_input,
4591 current_input = tuple(ta.read(time) for ta in input_ta)
4592 current_input = nest.pack_sequence_as(inputs, current_input)
4593 output, new_states = step_function(current_input,
6280 def range_less_than(old_input, current_input): argument
6283 max_num_labels_tns, current_input)
|