Home | History | Annotate | Download | only in lib

Lines Matching refs:inp

264       non_control_inputs = [inp for inp in curr_node.inputs]
265 control_inputs = [inp for inp in curr_node.control_inputs]
270 for inp in all_inputs:
271 all_input_nodes.add(self._get_node(inp).name)
275 for inp in all_inputs:
278 if inp.name not in self._output_targets:
279 self._output_targets[inp.name] = set([curr_elem.name])
281 self._output_targets[inp.name].add(curr_elem.name)
283 if (isinstance(inp, ops.Tensor) and
284 inp.op.type in ["Variable", "VariableV2"] and
285 inp.name not in self._variable_initializers):
288 initializer = graph.as_graph_element(inp.op.name + "/Assign")
289 self._variable_initializers[inp.name] = initializer
290 self._variable_initial_values[inp.name] = initializer.inputs[1]
292 inp_node = self._get_node(inp)
297 elem_stack.append(inp)
298 closure_elements.append(inp.name)
299 if isinstance(inp, ops.Tensor) and inp.dtype._is_ref_dtype: # pylint: disable=protected-access
300 ref_tensor_names.add(inp.name)
570 non_control_inputs = [inp for inp in curr_node.inputs]
571 control_inputs = [inp for inp in curr_node.control_inputs]
575 for inp in all_inputs:
578 if isinstance(inp, ops.Tensor):
579 is_inp_ref = inp.dtype._is_ref_dtype # pylint: disable=protected-access
580 can_feed = self._sess.graph.is_feedable(inp) and not is_inp_ref
585 if (restore_variable_values and inp.name in self._dirty_variables and
586 inp.name not in restored_variables and
587 inp.name not in self._last_updated):
590 initializer_op = self._variable_initializers[inp.name]
591 initial_value_tensor = self._variable_initial_values[inp.name]
595 self._cached_variable_values[inp.name]
599 restored_variables.add(inp.name)
605 if (is_inp_ref and inp.op.type in ["Variable", "VariableV2"] and
608 self._last_updated.add(inp.name)
611 if inp.name not in self._cached_variable_values:
612 old_value = self._sess.run(inp)
613 self._cached_variable_values[inp.name] = old_value
620 if use_overrides and can_feed and inp.name in self._override_tensors:
622 feeds[inp] = self._override_tensors[inp.name]
623 self._last_feed_types[inp.name] = self.FEED_TYPE_OVERRIDE
624 elif (can_feed and inp not in feeds and
625 use_tensor_handles and inp.name in self._tensor_handles):
627 feeds[inp] = self._tensor_handles[inp.name]
628 self._last_feed_types[inp.name] = self.FEED_TYPE_HANDLE
629 elif (can_feed and inp not in feeds and
631 inp.name in self._dumped_intermediate_tensors):
633 feeds[inp] = self._dumped_intermediate_tensors[inp.name].get_tensor()
634 self._last_feed_types[inp.name] = self.FEED_TYPE_DUMPED_INTERMEDIATE
635 elif inp.name in self._client_feed_dict:
637 feeds[inp] = self._client_feed_dict[inp.name]
638 self._last_feed_types[inp.name] = self.FEED_TYPE_CLIENT
642 inp_node = self._get_node(inp)
647 elem_stack.append(inp)