Home
last modified time | relevance | path

Searched defs:expected_dtype (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dconvert_to_tensor_or_ragged_tensor_op_test.py117 expected_dtype=None): argument
205 expected_dtype=None): argument
Dragged_constant_value_op_test.py189 expected_dtype=None): argument
Dragged_const_op_test.py180 expected_dtype=None): argument
Dragged_dispatch_test.py221 expected_dtype=None, argument
Drow_partition_test.py944 expected_uniform_row_length, expected_dtype): argument
Ddynamic_ragged_shape_test.py2627 def testAddingRowSplits(self, left_dtype, right_dtype, expected_dtype): argument
/external/tensorflow/tensorflow/python/training/
Dtraining_util_test.py27 def _assert_global_step(self, global_step, expected_dtype=dtypes.int64): argument
/external/tensorflow/tensorflow/python/framework/
Dpython_tensor_converter.cc56 bool CheckDType(DataType actual_dtype, DataType& expected_dtype) { in CheckDType()
Dtest_util.py3401 def assertDTypeEqual(self, target, expected_dtype): argument
/external/python/pybind11/tests/
Dtest_numpy_dtypes.py80 def assert_equal(actual, expected_data, expected_dtype): argument
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dsignature_flattening_test.cc33 DataType expected_dtype) { in TensorSpecsAreEqual()
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer_test.py64 def create_identity_with_grad_check_fn(expected_gradient, expected_dtype=None): argument
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dtest_util.py29 def create_identity_with_grad_check_fn(expected_gradient, expected_dtype=None): argument
/external/tensorflow/tensorflow/cc/ops/
Dconst_op_test.cc38 void ExpectTypeAndShape(const Node* n, DataType expected_dtype, in ExpectTypeAndShape()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dcsr_sparse_matrix_to_sparse_tensor_op.cc47 DataType expected_dtype) { in ValidateCSRSparseMatrix()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops.cc121 DataType expected_dtype) { in CreateResourceInvalidDTypeError()
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.cc559 DataType expected_dtype = RemoveRefType(input_types[inputs->size()]); in AddDatasetInput() local
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device.cc788 int expected_rank, TF_DataType* expected_dtype, in VerifyTensorRankAndDType()