Searched defs:expected_dtype (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | convert_to_tensor_or_ragged_tensor_op_test.py | 121 expected_dtype=None): argument
209 expected_dtype=None): argument
|
| D | ragged_constant_value_op_test.py | 188 expected_dtype=None): argument
|
| D | ragged_const_op_test.py | 184 expected_dtype=None): argument
|
| D | row_partition_test.py | 692 expected_uniform_row_length, expected_dtype): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | training_util_test.py | 31 def _assert_global_step(self, global_step, expected_dtype=dtypes.int64): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_tensor_converter.cc | 56 bool CheckDType(DataType actual_dtype, DataType& expected_dtype) { in CheckDType()
|
| D | test_util.py | 3041 def assertDTypeEqual(self, target, expected_dtype): argument
|
| /external/python/pybind11/tests/ |
| D | test_numpy_dtypes.py | 80 def assert_equal(actual, expected_data, expected_dtype): argument
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | signature_flattening_test.cc | 33 DataType expected_dtype) { in TensorSpecsAreEqual()
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer_test.py | 69 def create_identity_with_grad_check_fn(expected_gradient, expected_dtype=None): argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | test_util.py | 32 def create_identity_with_grad_check_fn(expected_gradient, expected_dtype=None): argument
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | const_op_test.cc | 38 void ExpectTypeAndShape(const Node* n, DataType expected_dtype, in ExpectTypeAndShape()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 47 DataType expected_dtype) { in ValidateCSRSparseMatrix()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 554 DataType expected_dtype = RemoveRefType(input_types[inputs->size()]); in AddDatasetInput() local
|