/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)
|