Searched defs:expected_dtype (Results 1 – 10 of 10) sorted by relevance
121 expected_dtype=None): argument209 expected_dtype=None): argument
188 expected_dtype=None): argument
184 expected_dtype=None): argument
33 def _assert_global_step(self, global_step, expected_dtype=dtypes.int64): argument
64 def assert_equal(actual, expected_data, expected_dtype): argument
33 def create_identity_with_grad_check_fn(expected_gradient, expected_dtype=None): argument
38 void ExpectTypeAndShape(const Node* n, DataType expected_dtype, in ExpectTypeAndShape()
47 DataType expected_dtype) { in ValidateCSRSparseMatrix()
546 DataType expected_dtype = RemoveRefType(input_types[inputs->size()]); in AddDatasetInput() local
2788 def assertDTypeEqual(self, target, expected_dtype): argument