Searched refs:structured_outputs (Results 1 – 16 of 16) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
D | while_v2_indexed_slices_rewriter.py | 67 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)
|
D | cond_v2.py | 311 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 …]
|
D | functional_ops.py | 849 then_out = then_branch.structured_outputs 850 else_out = else_branch.structured_outputs
|
D | while_v2.py | 673 grad_func_graph.structured_outputs.append(new_output) 780 output = body_grad_graph.structured_outputs[structured_outputs_idx]
|
/external/tensorflow/tensorflow/python/eager/ |
D | function.cc | 32 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()
|
D | function_argument_naming_test.py | 57 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()))
|
D | wrap_function.py | 371 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,
|
D | function.py | 744 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/ |
D | def_function.py | 51 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/ |
D | function_serialization.py | 76 structured_outputs = func_graph_module.convert_structure_to_signature( 77 concrete_function.structured_outputs) 81 coder.encode_structure(structured_outputs))
|
D | signature_serialization.py | 61 _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)
|
D | load.py | 327 …concrete_function._func_graph.structured_outputs = original_outputs # pylint: disable=protected-a…
|
/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
D | trt_convert.py | 1009 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)
|
D | trt_convert_test.py | 665 return root.signatures[signature_key].structured_outputs
|
/external/tensorflow/tensorflow/python/framework/ |
D | func_graph.py | 196 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/ |
D | cluster_coordinator.py | 372 self._concrete_function.structured_outputs) 1353 self._dataset_fn.structured_outputs.element_spec))
|