Home
last modified time | relevance | path

Searched refs:structured_outputs (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dwhile_v2_indexed_slices_rewriter.py67 structured_outputs = body_grad_graph.structured_outputs[3:]
69 for forward_input, output in zip(inputs_with_grads, structured_outputs):
94 idx = body_grad_graph.structured_outputs.index(grad_output_slices)
95 body_grad_graph.structured_outputs[idx] = new_output
97 body_grad_graph.structured_outputs)
262 structured_idx = graph.structured_outputs.index(old_output_slices)
268 graph.structured_outputs[structured_idx] = output_slices
270 graph.structured_outputs)
Dcond_v2.py222 return func_graph_module.pack_sequence_as(true_graph.structured_outputs,
496 true_outputs = list(true_graph.structured_outputs)
497 false_outputs = list(false_graph.structured_outputs)
517 true_graph.structured_outputs = true_outputs
519 false_graph.structured_outputs = false_outputs
576 "%s" % (true_graph.structured_outputs, false_graph.structured_outputs,
580 nest.assert_same_structure(true_graph.structured_outputs,
581 false_graph.structured_outputs,
Dwhile_v2.py470 grad_func_graph.structured_outputs.append(new_output)
577 output = body_grad_graph.structured_outputs[structured_outputs_idx]
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_serialization.py56 structured_outputs = func_graph_module.convert_structure_to_signature(
57 concrete_function.structured_outputs)
61 coder.encode_structure(structured_outputs))
Dsignature_serialization.py55 _normalize_outputs(concrete_function.structured_outputs, "unused", "unused")
107 structured_outputs = signature_function(**kwargs)
109 structured_outputs, signature_function.name, signature_key)
237 assert isinstance(func.structured_outputs, collections.Mapping)
Dload.py89 …concrete_function._func_graph.structured_outputs = original_outputs # pylint: disable=protected-a…
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py99 output_graph.structured_outputs = input_func.graph.structured_outputs
Dfunc_graph.py184 self.structured_outputs = None
724 func_graph.structured_outputs = func_outputs
728 for x in flatten(func_graph.structured_outputs)
/external/tensorflow/tensorflow/python/eager/
Dfunction.py685 def structured_outputs(self): member in ConcreteFunction
687 return self._func_graph.structured_outputs
708 outputs_list = nest.flatten(self._func_graph.structured_outputs)
722 return nest.pack_sequence_as(self._func_graph.structured_outputs,
729 self._func_graph.structured_outputs)
797 backwards_graph.structured_outputs = gradients_wrt_inputs
902 if self._func_graph.structured_outputs is None:
907 outputs_list = nest.flatten(self._func_graph.structured_outputs)
926 ret = nest.pack_sequence_as(self._func_graph.structured_outputs,
Dfunction_argument_naming_test.py57 self.assertEqual(2, len(fn_op.graph.structured_outputs))
81 self.assertEqual(2, len(fn_op.graph.structured_outputs))
101 set(fn_op.graph.structured_outputs.keys()))
Dwrap_function.py206 pruned_graph.structured_outputs = nest.map_structure(
290 fn_outputs = self.graph.structured_outputs
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py302 converted_graph.structured_outputs = func.graph.structured_outputs