Home
last modified time | relevance | path

Searched refs:structured_outputs (Results 1 – 16 of 16) 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):
102 idx = _get_tensor_index_in_iterable(body_grad_graph.structured_outputs,
104 body_grad_graph.structured_outputs[idx] = new_output
106 body_grad_graph.structured_outputs)
271 structured_idx = _get_tensor_index_in_iterable(graph.structured_outputs,
280 graph.structured_outputs[structured_idx] = output_slices
282 graph.structured_outputs)
Dcond_v2.py311 return _pack_sequence_as(true_graph.structured_outputs, tensors)
562 branch_outputs = [g.structured_outputs for g in grad_graphs]
573 grad_graphs[branch_index].structured_outputs[output_idx] = zeros
577 t for t in func_graph_module.flatten(grad_graph.structured_outputs)
601 branch_outputs = [g.structured_outputs for g in branch_graphs]
626 branch_graph.structured_outputs = branch_outs
642 nest.flatten(branch_graph.structured_outputs, expand_composites=False)
692 branch_graph.structured_outputs = _pack_sequence_as(
693 branch_graph.structured_outputs, branch_graph.outputs)
696 def _pack_sequence_as(structured_outputs, op_outputs): argument
[all …]
Dfunctional_ops.py849 then_out = then_branch.structured_outputs
850 else_out = else_branch.structured_outputs
Dwhile_v2.py673 grad_func_graph.structured_outputs.append(new_output)
780 output = body_grad_graph.structured_outputs[structured_outputs_idx]
/external/tensorflow/tensorflow/python/eager/
Dfunction.cc32 py::object structured_outputs,
37 py::object result, py::object structured_outputs, bool _ndarrays_list, in _build_call_outputs() argument
45 if (structured_outputs.is_none()) { in _build_call_outputs()
66 py::list outputs_list = nest->attr("flatten")(structured_outputs, true); in _build_call_outputs()
74 return nest->attr("pack_sequence_as")(structured_outputs, outputs_list, true); in _build_call_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.py371 pruned_graph.structured_outputs = nest.map_structure(
546 flat_fn_outputs = nest.flatten(self.graph.structured_outputs)
549 fn_outputs = nest.pack_sequence_as(self.graph.structured_outputs,
Dfunction.py744 return backwards_function.structured_outputs
959 backwards_graph.structured_outputs = gradients_wrt_inputs
1159 wrapped_backwards_graph.structured_outputs = dinputs
1256 return backward.structured_outputs
2022 def structured_outputs(self): member in ConcreteFunction
2024 return self._func_graph.structured_outputs
2047 self._func_graph.structured_outputs),
2056 self._func_graph.structured_outputs),
2180 if self._func_graph.structured_outputs is None:
2185 self._func_graph.structured_outputs, expand_composites=True)
[all …]
/external/tensorflow/tensorflow/python/framework/experimental/
Ddef_function.py51 structured_outputs = self._python_func(*structured_args,
54 py_outputs = nest.flatten(structured_outputs, expand_composites=True)
70 return nest.pack_sequence_as(structured_outputs, call_op_outputs)
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_serialization.py76 structured_outputs = func_graph_module.convert_structure_to_signature(
77 concrete_function.structured_outputs)
81 coder.encode_structure(structured_outputs))
Dsignature_serialization.py61 _normalize_outputs(concrete_function.structured_outputs, "unused", "unused")
156 structured_outputs = signature_function(**kwargs)
158 structured_outputs, signature_function.name, signature_key)
285 assert isinstance(func.structured_outputs, collections_abc.Mapping)
Dload.py327 …concrete_function._func_graph.structured_outputs = original_outputs # pylint: disable=protected-a…
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py1009 rebuilt_func.graph.structured_outputs = nest.pack_sequence_as(
1010 func.graph.structured_outputs, rebuilt_func.graph.structured_outputs)
1064 self._converted_func.graph.structured_outputs = nest.pack_sequence_as(
1065 func.graph.structured_outputs,
1066 self._converted_func.graph.structured_outputs)
1224 reset_converted_func.graph.structured_outputs = nest.pack_sequence_as(
1225 self._converted_func.graph.structured_outputs,
1226 reset_converted_func.graph.structured_outputs)
Dtrt_convert_test.py665 return root.signatures[signature_key].structured_outputs
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py196 self.structured_outputs = None
1032 func_graph.structured_outputs = func_outputs
1036 for x in flatten(func_graph.structured_outputs)
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator.py372 self._concrete_function.structured_outputs)
1353 self._dataset_fn.structured_outputs.element_spec))