| /external/tensorflow/tensorflow/python/framework/ |
| D | composite_tensor_test.py | 49 def assertNestEqual(self, a, b, expand_composites=False): argument
60 elif expand_composites and isinstance(a, composite_tensor.CompositeTensor):
69 x = nest.flatten(structure, expand_composites=True)
85 result = nest.pack_sequence_as(structure1, flat, expand_composites=True)
93 nest.assert_same_structure(st1, st2, expand_composites=False)
94 nest.assert_same_structure(st1, st2, expand_composites=True)
95 nest.assert_same_structure(st1, test, expand_composites=False)
97 nest.assert_same_structure(st1, test, expand_composites=True)
|
| D | func_graph.py | 799 flat_sequence = nest.flatten(sequence, expand_composites=True)
820 flattened_structure = nest.flatten(structure, expand_composites=True)
827 return nest.pack_sequence_as(structure, flat_sequence, expand_composites=True)
|
| /external/tensorflow/tensorflow/python/util/ |
| D | nest.py | 202 def flatten(structure, expand_composites=False): argument
232 return _pywrap_tensorflow.Flatten(structure, expand_composites)
253 expand_composites=False): argument
288 expand_composites)
387 def pack_sequence_as(structure, flat_sequence, expand_composites=False): argument
419 is_seq = is_sequence_or_composite if expand_composites else is_sequence
490 expand_composites = False
493 expand_composites = kwargs.pop("expand_composites", expand_composites)
500 expand_composites=expand_composites)
502 flat_structure = [flatten(s, expand_composites) for s in structure]
[all …]
|
| D | util.h | 146 bool expand_composites);
177 PyObject* Flatten(PyObject* nested, bool expand_composites = false);
|
| D | util.cc | 795 PyObject* Flatten(PyObject* nested, bool expand_composites) { in Flatten() argument
798 expand_composites ? IsSequenceOrCompositeHelper : IsSequenceHelper; in Flatten()
800 expand_composites ? GetValueIteratorForComposite : GetValueIterator; in Flatten()
914 bool expand_composites) { in AssertSameStructure() argument
916 expand_composites ? IsSequenceOrCompositeHelper : IsSequenceHelper; in AssertSameStructure()
918 expand_composites ? GetValueIteratorForComposite : GetValueIterator; in AssertSameStructure()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nest.pbtxt | 5 …argspec: "args=[\'nest1\', \'nest2\', \'check_types\', \'expand_composites\'], varargs=None, keywo…
9 …argspec: "args=[\'structure\', \'expand_composites\'], varargs=None, keywords=None, defaults=[\'Fa…
21 …argspec: "args=[\'structure\', \'flat_sequence\', \'expand_composites\'], varargs=None, keywords=N…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nest.pbtxt | 5 …argspec: "args=[\'nest1\', \'nest2\', \'check_types\', \'expand_composites\'], varargs=None, keywo…
9 …argspec: "args=[\'structure\', \'expand_composites\'], varargs=None, keywords=None, defaults=[\'Fa…
21 …argspec: "args=[\'structure\', \'flat_sequence\', \'expand_composites\'], varargs=None, keywords=N…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 191 return nest.map_structure(_Identity, data, expand_composites=True)
204 return nest.map_structure(_NextIteration, data, expand_composites=True)
250 return nest.map_structure(enter_component, data, expand_composites=True)
274 return nest.map_structure(exit, data, expand_composites=True)
307 tensors = nest.flatten(data, expand_composites=True)
310 return (nest.pack_sequence_as(data, mapped_f, expand_composites=True),
311 nest.pack_sequence_as(data, mapped_t, expand_composites=True))
406 nest.assert_same_structure(inputs[0], v, expand_composites=True)
408 flat_inputs = [nest.flatten(v, expand_composites=True) for v in inputs]
414 expand_composites=True)
[all …]
|
| D | while_v2_indexed_slices_rewriter.py | 279 return nest.flatten(arg, expand_composites=True)
|
| D | cond_v2.py | 582 expand_composites=True)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 795 for grad in nest.flatten(gradients_wrt_inputs, expand_composites=True)
|