| /external/tensorflow/tensorflow/python/data/util/ |
| sparse.py | 20 from tensorflow.python.data.util import nest
37 return any(c is sparse_tensor.SparseTensor for c in nest.flatten(classes))
52 ret = nest.pack_sequence_as(shapes, [
54 for shape, c in zip(nest.flatten(shapes), nest.flatten(classes))
71 ret = nest.pack_sequence_as(types, [
73 for ty, c in zip(nest.flatten(types), nest.flatten(classes))
91 ret = nest.pack_sequence_as(types, [
95 nest.flatten(tensors), nest.flatten(types), nest.flatten(shapes)
[all...] |
| nest_test.py | 25 from tensorflow.python.data.util import nest
38 self.assertEqual(nest.flatten(structure), [3, 4, 5, 6, 7, 9, 10, 8])
40 nest.pack_sequence_as(structure, flat), (("a", "b"), "c",
45 self.assertEqual(nest.flatten(structure), flat)
46 restructured_from_flat = nest.pack_sequence_as(structure, flat)
53 self.assertEqual([5], nest.flatten(5))
54 self.assertEqual([np.array([5])], nest.flatten(np.array([5])))
56 self.assertEqual("a", nest.pack_sequence_as(5, ["a"]))
58 np.array([5]), nest.pack_sequence_as("scalar", [np.array([5])]))
61 nest.pack_sequence_as("scalar", [4, 5]
[all...] |
| sparse_test.py | 21 from tensorflow.python.data.util import nest
75 for a, b in zip(nest.flatten(a), nest.flatten(b)):
320 shapes = nest.map_structure(lambda _: tensor_shape.TensorShape(None),
322 types = nest.map_structure(lambda _: dtypes.int32, classes)
326 nest.assert_same_structure(expected, actual)
327 for a, e in zip(nest.flatten(actual), nest.flatten(expected)):
350 shapes = nest.map_structure(lambda _: tensor_shape.TensorShape(None),
352 types = nest.map_structure(lambda _: dtypes.int32, classes
[all...] |
| structure.py | 24 from tensorflow.python.data.util import nest
81 `tf.contrib.framework.nest.assert_same_structure`, and each nested
271 flat_types = nest.flatten(output_types)
272 flat_shapes = nest.flatten(output_shapes)
273 flat_classes = nest.flatten(output_classes)
290 ret = nest.pack_sequence_as(output_classes, flat_ret)
306 self._flat_nested_structure = nest.flatten(nested_structure)
309 for s in nest.flatten(nested_structure):
329 nest.assert_same_structure(self._nested_structure,
337 nest.flatten(self._nested_structure)
[all...] |
| /external/tensorflow/tensorflow/python/util/ |
| nest_test.py | 34 from tensorflow.python.util import nest
78 self.assertFalse(nest._is_attrs(field_values))
79 self.assertTrue(nest._is_attrs(sample_attr))
80 flat = nest.flatten(sample_attr)
82 restructured_from_flat = nest.pack_sequence_as(sample_attr, flat)
88 flat = nest.flatten(NestTest.BadAttr())
94 self.assertEqual(nest.flatten(structure), [3, 4, 5, 6, 7, 9, 10, 8])
96 nest.pack_sequence_as(structure, flat), (("a", "b"), "c",
101 self.assertEqual(nest.flatten(structure), flat)
102 restructured_from_flat = nest.pack_sequence_as(structure, flat
[all...] |
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| script_ops.py | 28 from tensorflow.python.util import nest
113 flat_output_types = nest.flatten(output_types)
115 flat_args = nest.flatten(args)
118 py_args, py_kwargs = nest.pack_sequence_as(args, py_args)
124 nest.assert_shallow_structure(output_types, ret)
125 return nest.flatten(ret)
136 output_shapes = nest.map_structure_up_to(
139 flattened_shapes = nest.flatten(output_shapes)
143 return nest.pack_sequence_as(output_types, flat_values)
|
| /external/libnetfilter_conntrack/src/conntrack/ |
| build_mnl.c | 19 struct nlattr *nest; local
21 nest = mnl_attr_nest_start(nlh, CTA_TUPLE_IP);
22 if (nest == NULL)
37 mnl_attr_nest_cancel(nlh, nest);
40 mnl_attr_nest_end(nlh, nest);
47 struct nlattr *nest; local
49 nest = mnl_attr_nest_start(nlh, CTA_TUPLE_PROTO);
50 if (nest == NULL)
76 mnl_attr_nest_cancel(nlh, nest);
79 mnl_attr_nest_end(nlh, nest);
97 struct nlattr *nest; local
116 struct nlattr *nest, *nest_proto; local
236 struct nlattr *nest; local
249 struct nlattr *nest; local
288 struct nlattr *nest; local
300 struct nlattr *nest; local
311 struct nlattr *nest; local
322 struct nlattr *nest; local
334 struct nlattr *nest; local
346 struct nlattr *nest; local
357 struct nlattr *nest; local
368 struct nlattr *nest; local
407 struct nlattr *nest; local
462 struct nlattr *nest; local
492 struct nlattr *nest; local
[all...] |
| build.c | 16 struct nfattr *nest; local
18 nest = nfnl_nest(&req->nlh, size, CTA_TUPLE_IP);
37 nfnl_nest_end(&req->nlh, nest);
44 struct nfattr *nest; local
46 nest = nfnl_nest(&req->nlh, size, CTA_TUPLE_PROTO);
86 nfnl_nest_end(&req->nlh, nest);
99 struct nfattr *nest; local
101 nest = nfnl_nest(&req->nlh, size, type);
103 nfnl_nest_end(&req->nlh, nest);
109 struct nfattr *nest, *nest_proto local
246 struct nfattr *nest; local
260 struct nfattr *nest; local
300 struct nfattr *nest; local
312 struct nfattr *nest; local
323 struct nfattr *nest; local
334 struct nfattr *nest; local
346 struct nfattr *nest; local
358 struct nfattr *nest; local
369 struct nfattr *nest; local
380 struct nfattr *nest; local
420 struct nfattr *nest; local
493 struct nfattr *nest; local
513 struct nfattr *nest; local
[all...] |
| /external/clang/test/PCH/ |
| struct.h | 28 struct Nested { int x, y; } nest; member in struct:S
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| batching.py | 23 from tensorflow.python.data.util import nest
201 flat_original_shapes = nest.flatten(original_shapes)
202 flat_new_shapes = nest.flatten_up_to(original_shapes, expected_shapes)
207 return nest.pack_sequence_as(original_shapes, flat_merged_output_shapes)
210 flatten_tensors = nest.flatten(elements)
211 flatten_shapes = nest.flatten(expected_shapes)
216 return nest.pack_sequence_as(elements, checked_tensors)
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| scan_ops.py | 23 from tensorflow.python.data.util import nest
41 self._initial_state = nest.pack_sequence_as(initial_state, [
45 for i, t in enumerate(nest.flatten(initial_state))
76 nest.flatten(new_state_classes),
77 nest.flatten(old_state_classes)):
88 nest.flatten(new_state_types), nest.flatten(old_state_types)):
101 flat_state_shapes = nest.flatten(old_state_shapes)
102 flat_new_state_shapes = nest.flatten(new_state_shapes)
123 nest.pack_sequence_as(old_state_shapes, weakened_state_shapes)
[all...] |
| /external/tensorflow/tensorflow/python/keras/saving/ |
| saving_utils.py | 23 from tensorflow.python.util import nest
75 flat_inputs = nest.flatten(inputs)
76 flat_input_names = nest.flatten(input_names)
82 input_specs = nest.pack_sequence_as(structure=inputs,
95 outputs_list = nest.flatten(model(inputs=inputs))
|
| /build/kati/testcase/ |
| nested_define.mk | 10 # Prefixed defines don't increase the nest level.
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| test_util.py | 27 from tensorflow.python.util import nest
34 targets1 = nest.flatten(targets1)
35 targets2 = ([] if targets2 is None else nest.flatten(targets2))
44 outputs = nest.flatten(outputs) # flatten SparseTensorValues
|
| control_flow_ops.py | 35 from tensorflow.python.util import nest
61 flat_loop_fn_dtypes = nest.flatten(loop_fn_dtypes)
66 fn_output = nest.flatten(loop_fn(i))
97 return nest.pack_sequence_as(loop_fn_dtypes, output)
211 for loop_fn_output in nest.flatten(loop_fn_outputs):
213 return nest.pack_sequence_as(loop_fn_outputs, outputs)
226 flattened_loop_fn_outputs = nest.flatten(loop_fn_outputs)
239 return nest.flatten(loop_fn(i + offset, pfor_config=pfor_config))
241 return nest.flatten(loop_fn(i + offset))
260 return nest.pack_sequence_as(loop_fn_outputs, nest.flatten(outputs)
[all...] |
| /external/iproute2/tipc/ |
| peer.c | 32 struct nlattr *nest; local
51 nest = mnl_attr_nest_start(nlh, TIPC_NLA_NET);
53 mnl_attr_nest_end(nlh, nest);
|
| /external/tensorflow/tensorflow/python/ops/ |
| rnn.py | 36 from tensorflow.python.util import nest
116 elif nest.is_sequence(state):
117 inferred_dtypes = [element.dtype for element in nest.flatten(state)]
227 flat_state = nest.flatten(state)
228 flat_zero_output = nest.flatten(zero_output)
259 nest.assert_same_structure(state, new_state)
261 flat_new_state = nest.flatten(new_state)
262 flat_new_output = nest.flatten(new_output)
277 nest.assert_same_structure(state, new_state)
278 new_state = nest.flatten(new_state
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| composite_tensor_test.py | 25 from tensorflow.python.util import nest
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)
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| iterator_ops.py | 25 from tensorflow.python.data.util import nest
180 output_types = nest.map_structure(dtypes.as_dtype, output_types)
182 output_shapes = nest.map_structure(
185 output_shapes = nest.map_structure_up_to(
188 output_classes = nest.map_structure(lambda _: ops.Tensor, output_types)
189 nest.assert_same_structure(output_types, output_shapes)
268 output_types = nest.map_structure(dtypes.as_dtype, output_types)
270 output_shapes = nest.map_structure(
273 output_shapes = nest.map_structure_up_to(
276 output_classes = nest.map_structure(lambda _: ops.Tensor, output_types
[all...] |
| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| recurrent.py | 58 from tensorflow.python.util import nest
82 return nest.map_structure(lambda x: array_ops.gather(x, index), struct)
101 acc_lst = nest.flatten(struct_acc)
102 x_lst = nest.flatten(struct_x)
113 return nest.pack_sequence_as(struct_acc, lst)
127 xs = nest.flatten(struct)
135 return nest.flatten(struct)
151 if not nest.is_sequence(elements):
152 return nest.pack_sequence_as(struct_template, [elements])
153 return nest.pack_sequence_as(struct_template, elements
[all...] |
| functional_rnn.py | 30 from tensorflow.python.util import nest
35 for x in nest.flatten(struct):
42 as_list = nest.flatten(struct)
43 template_as_list = nest.flatten(struct_template)
66 like_inputs_t = nest.map_structure(
73 inputs_t, state0 = nest.pack_sequence_as(input_structure, flat_inputs)
77 state_list = nest.flatten(state1)
129 extended_state0_flat = nest.flatten(extended_state0)
171 return nest.flatten(state)[self._output_state_idx]
202 for state_var in nest.flatten(acc_state)
[all...] |
| /external/tensorflow/tensorflow/python/feature_column/ |
| utils.py | 27 from tensorflow.python.util import nest
106 if nest.is_sequence(default_value):
115 isinstance(v, int) for v in nest.flatten(default_value))
117 isinstance(v, float) for v in nest.flatten(default_value))
135 if not nest.is_sequence(value):
145 if nest.is_sequence(default_value) != bool(shape):
|
| /external/iproute2/tc/ |
| q_prio.c | 36 struct rtattr *nest; local
90 nest = addattr_nest_compat(n, 1024, TCA_OPTIONS, &opt, sizeof(opt));
93 addattr_nest_compat_end(n, nest);
|
| q_rr.c | 37 struct rtattr *nest; local
85 nest = addattr_nest_compat(n, 1024, TCA_OPTIONS, &opt, sizeof(opt));
88 addattr_nest_compat_end(n, nest);
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| basic_decoder.py | 32 from tensorflow.python.util import nest
91 output_shape_with_unknown_batch = nest.map_structure(
96 return nest.map_structure(lambda s: s[1:], layer_output_shape)
110 dtype = nest.flatten(self._initial_state)[0].dtype
112 nest.map_structure(lambda _: dtype, self._rnn_output_size()),
186 self._cell_dtype = nest.flatten(initial_state)[0].dtype
204 output_shape_with_unknown_batch = nest.map_structure(
208 return nest.map_structure(lambda s: s[1:], layer_output_shape)
224 nest.map_structure(lambda _: dtype, self._rnn_output_size()),
|
