Home
last modified time | relevance | path

Searched refs:assert_same_structure (Results 1 – 25 of 43) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dnest_test.py195 nest.assert_same_structure(structure1, structure2)
196 nest.assert_same_structure("abc", 1.0)
197 nest.assert_same_structure("abc", np.array([0, 1]))
198 nest.assert_same_structure("abc", constant_op.constant([0, 1]))
202 nest.assert_same_structure(structure1, structure_different_num_elements)
206 nest.assert_same_structure((0, 1), np.array([0, 1]))
210 nest.assert_same_structure(0, (0, 1))
214 nest.assert_same_structure(structure1, structure_different_nesting)
218 self.assertRaises(TypeError, nest.assert_same_structure, (0, 1),
221 nest.assert_same_structure(named_type_0(3, 4), named_type_0("a", "b"))
[all …]
H A Dnest.py82 def assert_same_structure(nest1, nest2, check_types=True): function
220 assert_same_structure(structure[0], other, check_types=check_types)
H A Dstructure.py338 nest.assert_same_structure(element_spec, element)
414 nest.assert_same_structure(spec1, spec2)
H A Dsparse_test.py321 nest.assert_same_structure(test_case, actual)
337 nest.assert_same_structure(test_case, actual)
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dnest_test.py351 nest.assert_same_structure(structure1, structure2)
352 nest.assert_same_structure("abc", 1.0)
353 nest.assert_same_structure("abc", np.array([0, 1]))
354 nest.assert_same_structure("abc", constant_op.constant([0, 1]))
368 nest.assert_same_structure(structure1, structure_different_num_elements)
378 nest.assert_same_structure([0, 1], np.array([0, 1]))
388 nest.assert_same_structure(0, [0, 1])
390 self.assertRaises(TypeError, nest.assert_same_structure, (0, 1), [0, 1])
395 nest.assert_same_structure(structure1, structure_different_nesting)
397 self.assertRaises(TypeError, nest.assert_same_structure, (0, 1),
[all …]
H A Dnest.py477 def assert_same_structure(nest1, nest2, check_types=True, function
910 assert_same_structure(structure[0], other, check_types=check_types,
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Drnn.py243 nest.assert_same_structure(zero_output, new_output)
244 nest.assert_same_structure(state, new_state)
263 nest.assert_same_structure(zero_output, new_output)
264 nest.assert_same_structure(state, new_state)
1188 nest.assert_same_structure(initial_state, cell.state_size)
1247 nest.assert_same_structure(state, cell_state)
1248 nest.assert_same_structure(cell.output_size, next_output)
1255 nest.assert_same_structure(state, next_state)
1256 nest.assert_same_structure(current_input, next_input)
1257 nest.assert_same_structure(emit_ta, emit_output)
H A Dfunctional_ops.py647 nest.assert_same_structure(elems if initializer is None else initializer,
853 nest.assert_same_structure(then_out, else_out, expand_composites=True)
H A Dcontrol_flow_ops.py429 nest.assert_same_structure(inputs[0], v, expand_composites=True)
1249 nest.assert_same_structure(orig_res_t, orig_res_f, expand_composites=True)
1254 nest.assert_same_structure(orig_res_t, orig_res_f,
2219 nest.assert_same_structure(
2766 nest.assert_same_structure(loop_var_structure, list(loop_vars))
H A Dmap_fn.py489 nest.assert_same_structure(fn_output_signature or elems, result_value)
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dtraining_utils.py89 nest.assert_same_structure(
92 nest.assert_same_structure(
96 nest.assert_same_structure(
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dcomposite_tensor_test.py202 nest.assert_same_structure(s1, s2, expand_composites=expand_composites)
223 nest.assert_same_structure(s1, s2, expand_composites=False)
226 nest.assert_same_structure(s1, s2, expand_composites=True)
406 nest.assert_same_structure(value, spec, expand_composites=True)
/aosp_15_r20/external/tensorflow/tensorflow/python/trackable/
H A Ddata_structures_test.py177 nest.assert_same_structure(l, data_structures.ListWrapper(copy.copy(l)))
499 nest.assert_same_structure(d, data_structures._DictWrapper(d.copy()))
660 nest.assert_same_structure(
682 nest.assert_same_structure(t, m.t)
683 nest.assert_same_structure(m.t, t)
688 nest.assert_same_structure(m.nt, nt)
690 nest.assert_same_structure(m.nt, m.t)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Diterator_ops.py228 nest.assert_same_structure(output_types, output_shapes)
302 nest.assert_same_structure(output_types, output_shapes)
365 nest.assert_same_structure(self.output_types, dataset_output_types)
366 nest.assert_same_structure(self.output_shapes, dataset_output_shapes)
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/
H A Dcontrol_flow.py291 nest.assert_same_structure(init, entry, expand_composites=True)
299 nest.assert_same_structure(init_tensors, entry, expand_composites=True)
305 nest.assert_same_structure(entry, exit_, expand_composites=True)
311 nest.assert_same_structure(init, invariant, expand_composites=False)
371 nest.assert_same_structure(body_var, orelse_var, expand_composites=True)
380 nest.assert_same_structure(body_var_tensors, orelse_var_tensors,
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding_v2.py445 nest.assert_same_structure(
678 nest.assert_same_structure(self._feature_config, gradients)
1232 nest.assert_same_structure(self._feature_config, features)
1252 nest.assert_same_structure(self._feature_config, weights)
1417 nest.assert_same_structure(self._output_shapes, incoming_output_shapes)
1431 nest.assert_same_structure(self._output_shapes, incoming_output_shapes)
H A Dtpu_embedding_for_serving.py278 nest.assert_same_structure(inputs, feature_config)
283 nest.assert_same_structure(inputs, weights)
H A Dtpu_embedding_v1.py260 nest.assert_same_structure(features, self._feature_config)
265 nest.assert_same_structure(features, weights)
H A Dtpu.py1314 nest.assert_same_structure(inputs[0], inputs[i])
1376 nest.assert_same_structure(inputs[0], maximum_shapes, check_types=False)
1395 nest.assert_same_structure(flat_inputs[0], flat_maximum_shapes,
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/
H A Dparameter_server_evaluation_test.py128 nest.assert_same_structure(
130 nest.assert_same_structure(
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dinput_lib_type_spec_test.py186 nest.assert_same_structure(element_spec, iter_element_spec)
313 nest.assert_same_structure(element_spec, iter_element_spec)
561 nest.assert_same_structure(element_spec, iter_element_spec)
751 nest.assert_same_structure(x, x._type_spec, expand_composites=True)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.nest.pbtxt4 name: "assert_same_structure"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.nest.pbtxt4 name: "assert_same_structure"
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dfrom_list.py43 nest.assert_same_structure(type_specs[0], type_specs[i])
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/experimental/rpc/
H A Drpc_ops.py375 nest.assert_same_structure(args, input_specs)

12