Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dwhile_v2_indexed_slices_rewriter.py65 structured_outputs = body_grad_graph.structured_outputs[3:]
67 for forward_input, output in zip(inputs_with_grads, structured_outputs):
101 idx = _get_tensor_index_in_iterable(body_grad_graph.structured_outputs,
103 body_grad_graph.structured_outputs[idx] = new_output
105 body_grad_graph.structured_outputs)
271 structured_idx = _get_tensor_index_in_iterable(graph.structured_outputs,
280 graph.structured_outputs[structured_idx] = output_slices
281 graph.outputs = func_graph.flatten(graph.structured_outputs)
Dcond_v2.py355 true_graph.structured_outputs,
356 false_graph.structured_outputs)
599 branch_outputs = [g.structured_outputs for g in grad_graphs]
610 grad_graphs[branch_index].structured_outputs[output_idx] = zeros
614 t for t in func_graph_module.flatten(grad_graph.structured_outputs)
638 branch_outputs = [g.structured_outputs for g in branch_graphs]
664 branch_graph.structured_outputs = branch_outs
680 nest.flatten(branch_graph.structured_outputs, expand_composites=False)
733 branch_graph.structured_outputs = _pack_sequence_as(
734 branch_graph.structured_outputs, branch_graph.outputs)
[all …]
Dbatch_ops.py117 computation.structured_outputs, outputs, expand_composites=True)
Dfunctional_ops.py848 then_out = then_branch.structured_outputs
849 else_out = else_branch.structured_outputs
Dwhile_v2.py722 grad_func_graph.structured_outputs.append(new_output)
829 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.py53 self.assertEqual(2, len(fn_op.graph.structured_outputs))
77 self.assertEqual(2, len(fn_op.graph.structured_outputs))
97 set(fn_op.graph.structured_outputs.keys()))
Dfunction.py649 return backwards_function.structured_outputs
863 backwards_graph.structured_outputs = gradients_wrt_inputs
1061 wrapped_backwards_graph.structured_outputs = dinputs
1158 return backward.structured_outputs
1913 def structured_outputs(self): member in ConcreteFunction
1915 return self._func_graph.structured_outputs
2052 self._func_graph.structured_outputs),
2061 self._func_graph.structured_outputs),
2185 if self._func_graph.structured_outputs is None:
2190 self._func_graph.structured_outputs, expand_composites=True)
[all …]
Dwrap_function.py369 pruned_graph.structured_outputs = nest.map_structure(
544 flat_fn_outputs = nest.flatten(self.graph.structured_outputs)
547 fn_outputs = nest.pack_sequence_as(self.graph.structured_outputs,
/external/tensorflow/tensorflow/python/framework/experimental/
Ddef_function.py47 structured_outputs = self._python_func(*structured_args,
50 py_outputs = nest.flatten(structured_outputs, expand_composites=True)
66 return nest.pack_sequence_as(structured_outputs, call_op_outputs)
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_serialization.py70 structured_outputs = func_graph_module.convert_structure_to_signature(
71 concrete_function.structured_outputs)
76 nested_structure_coder.encode_structure(structured_outputs))
Dsignature_serialization.py58 _normalize_outputs(concrete_function.structured_outputs, "unused", "unused")
157 structured_outputs = signature_function(**kwargs)
159 structured_outputs, signature_function.name, signature_key)
291 assert isinstance(func.structured_outputs, collections_abc.Mapping)
Dfunction_deserialization.py397 structured_outputs = None
407 structured_outputs = nested_structure_coder.decode_proto(
419 structured_outputs=structured_outputs)
Dtracing_utils.py62 concrete_save.structured_outputs)
Dtracing_utils_test.py54 restored_tensor_spec = t._serialize_to_tensors.structured_outputs
/external/tensorflow/tensorflow/python/framework/
Dfunction_def_to_graph.py36 structured_outputs=None, argument
65 structured_outputs=structured_outputs)
Dfunc_graph.py208 structured_outputs=None): argument
239 self.structured_outputs = structured_outputs
1293 func_graph.structured_outputs = func_outputs
1297 for x in flatten(func_graph.structured_outputs)
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dvalues.py365 self._coordinator.strategy, self._dataset_fn.structured_outputs))
379 return self._dataset_fn.structured_outputs.element_spec
Dcluster_coordinator.py227 self._concrete_function.structured_outputs)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py917 new_func.graph.structured_outputs = nest.pack_sequence_as(
918 func.graph.structured_outputs, new_func.graph.structured_outputs)
1578 reset_converted_func.graph.structured_outputs = nest.pack_sequence_as(
1579 self._converted_func.graph.structured_outputs,
1580 reset_converted_func.graph.structured_outputs)
Dtrt_convert_test.py713 return root.signatures[signature_key].structured_outputs
/external/tensorflow/tensorflow/python/distribute/experimental/rpc/
Drpc_ops.py39 func.structured_outputs)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.__internal__.-func-graph.pbtxt84 …ections\', \'capture_by_value\', \'structured_input_signature\', \'structured_outputs\'], varargs=…