Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/data/util/
Dnest_test.py199 nest.assert_same_structure(structure1, structure2)
200 nest.assert_same_structure("abc", 1.0)
201 nest.assert_same_structure("abc", np.array([0, 1]))
202 nest.assert_same_structure("abc", constant_op.constant([0, 1]))
206 nest.assert_same_structure(structure1, structure_different_num_elements)
210 nest.assert_same_structure((0, 1), np.array([0, 1]))
214 nest.assert_same_structure(0, (0, 1))
218 nest.assert_same_structure(structure1, structure_different_nesting)
222 self.assertRaises(TypeError, nest.assert_same_structure, (0, 1),
225 nest.assert_same_structure(named_type_0(3, 4), named_type_0("a", "b"))
[all …]
Dnest.py89 def assert_same_structure(nest1, nest2, check_types=True): function
224 assert_same_structure(structure[0], other, check_types=check_types)
Dstructure.py333 nest.assert_same_structure(element_spec, element)
409 nest.assert_same_structure(spec1, spec2)
Dsparse_test.py325 nest.assert_same_structure(test_case, actual)
341 nest.assert_same_structure(test_case, actual)
/external/tensorflow/tensorflow/python/util/
Dnest_test.py358 nest.assert_same_structure(structure1, structure2)
359 nest.assert_same_structure("abc", 1.0)
360 nest.assert_same_structure("abc", np.array([0, 1]))
361 nest.assert_same_structure("abc", constant_op.constant([0, 1]))
375 nest.assert_same_structure(structure1, structure_different_num_elements)
385 nest.assert_same_structure([0, 1], np.array([0, 1]))
395 nest.assert_same_structure(0, [0, 1])
397 self.assertRaises(TypeError, nest.assert_same_structure, (0, 1), [0, 1])
402 nest.assert_same_structure(structure1, structure_different_nesting)
404 self.assertRaises(TypeError, nest.assert_same_structure, (0, 1),
[all …]
Dnest.py440 def assert_same_structure(nest1, nest2, check_types=True, function
862 assert_same_structure(structure[0], other, check_types=check_types,
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils.py89 nest.assert_same_structure(
92 nest.assert_same_structure(
96 nest.assert_same_structure(
/external/tensorflow/tensorflow/python/ops/
Drnn.py246 nest.assert_same_structure(zero_output, new_output)
247 nest.assert_same_structure(state, new_state)
266 nest.assert_same_structure(zero_output, new_output)
267 nest.assert_same_structure(state, new_state)
1184 nest.assert_same_structure(initial_state, cell.state_size)
1243 nest.assert_same_structure(state, cell_state)
1244 nest.assert_same_structure(cell.output_size, next_output)
1251 nest.assert_same_structure(state, next_state)
1252 nest.assert_same_structure(current_input, next_input)
1253 nest.assert_same_structure(emit_ta, emit_output)
Dfunctional_ops.py651 nest.assert_same_structure(elems if initializer is None else initializer,
857 nest.assert_same_structure(then_out, else_out, expand_composites=True)
Dcontrol_flow_ops.py427 nest.assert_same_structure(inputs[0], v, expand_composites=True)
1308 nest.assert_same_structure(orig_res_t, orig_res_f, expand_composites=True)
1313 nest.assert_same_structure(orig_res_t, orig_res_f,
2261 nest.assert_same_structure(
2799 nest.assert_same_structure(loop_var_structure, list(loop_vars))
2816 nest.assert_same_structure(
Dmap_fn.py492 nest.assert_same_structure(fn_output_signature or elems, result_value)
/external/tensorflow/tensorflow/python/framework/
Dcomposite_tensor_test.py206 nest.assert_same_structure(s1, s2, expand_composites=expand_composites)
227 nest.assert_same_structure(s1, s2, expand_composites=False)
230 nest.assert_same_structure(s1, s2, expand_composites=True)
410 nest.assert_same_structure(value, spec, expand_composites=True)
Dfunc_graph.py315 nest.assert_same_structure(spec, ret_nest, expand_composites=True)
1092 nest.assert_same_structure(n1, n2, expand_composites=True)
/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py220 nest.assert_same_structure(output_types, output_shapes)
294 nest.assert_same_structure(output_types, output_shapes)
353 nest.assert_same_structure(self.output_types, dataset_output_types)
354 nest.assert_same_structure(self.output_shapes, dataset_output_shapes)
/external/tensorflow/tensorflow/python/keras/distribute/
Dparameter_server_evaluation_test.py128 nest.assert_same_structure(
130 nest.assert_same_structure(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nest.pbtxt4 name: "assert_same_structure"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nest.pbtxt4 name: "assert_same_structure"
/external/tensorflow/tensorflow/python/training/tracking/
Ddata_structures_test.py182 nest.assert_same_structure(l, data_structures.ListWrapper(copy.copy(l)))
504 nest.assert_same_structure(d, data_structures._DictWrapper(d.copy()))
665 nest.assert_same_structure(
687 nest.assert_same_structure(t, m.t)
688 nest.assert_same_structure(m.t, t)
693 nest.assert_same_structure(m.nt, nt)
695 nest.assert_same_structure(m.nt, m.t)
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_type_spec_test.py190 nest.assert_same_structure(element_spec, iter_element_spec)
317 nest.assert_same_structure(element_spec, iter_element_spec)
565 nest.assert_same_structure(element_spec, iter_element_spec)
755 nest.assert_same_structure(x, x._type_spec, expand_composites=True)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2.py535 nest.assert_same_structure(self._feature_config, gradients)
1262 nest.assert_same_structure(self._feature_config, features)
1267 nest.assert_same_structure(self._feature_config, weights)
1583 nest.assert_same_structure(inputs, feature_config)
1588 nest.assert_same_structure(inputs, weights)
Dtpu.py1295 nest.assert_same_structure(inputs[0], inputs[i])
1355 nest.assert_same_structure(inputs[0], maximum_shapes, check_types=False)
1374 nest.assert_same_structure(flat_inputs[0], flat_maximum_shapes,
/external/tensorflow/tensorflow/python/autograph/operators/
Dcontrol_flow.py293 nest.assert_same_structure(init, entry, expand_composites=True)
294 nest.assert_same_structure(entry, exit_, expand_composites=True)
300 nest.assert_same_structure(init, invariant, expand_composites=False)
359 nest.assert_same_structure(body_var, orelse_var, expand_composites=True)
Dcontrol_flow_deprecated_py2.py206 nest.assert_same_structure(entry, exit_, expand_composites=True)
212 nest.assert_same_structure(init, invariant, expand_composites=False)
267 nest.assert_same_structure(
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dtest_base.py148 nest.assert_same_structure(result_values[i], expected_values[i])
/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
Drnn_cell_wrapper_impl.py376 nest.assert_same_structure(inputs, outputs)

12