Home
last modified time | relevance | path

Searched refs:expand_composites (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dcomposite_tensor_test.py49 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)
Dfunc_graph.py799 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/
Dnest.py202 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 …]
Dutil.h146 bool expand_composites);
177 PyObject* Flatten(PyObject* nested, bool expand_composites = false);
Dutil.cc795 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/
Dtensorflow.nest.pbtxt5 …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/
Dtensorflow.nest.pbtxt5 …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/
Dcontrol_flow_ops.py191 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 …]
Dwhile_v2_indexed_slices_rewriter.py279 return nest.flatten(arg, expand_composites=True)
Dcond_v2.py582 expand_composites=True)
/external/tensorflow/tensorflow/python/eager/
Dfunction.py795 for grad in nest.flatten(gradients_wrt_inputs, expand_composites=True)