Searched defs:valid_dtypes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ | ||
| D | scan_ops_test.py | 75 valid_dtypes = [np.float32, np.int32, np.int64] variable in CumsumTest |
| /external/tensorflow/tensorflow/python/kernel_tests/ | ||
| D | scan_ops_test.py | 75 valid_dtypes = [ variable in CumsumTest |
| D | cumulative_logsumexp_test.py | 33 valid_dtypes = [dtypes.float32, dtypes.float64] variable in CumulativeLogsumexpTest |
| /external/tensorflow/tensorflow/cc/gradients/ | ||
| D | math_grad.cc | 1182 static constexpr DataType valid_dtypes[] = { in IsFloatingPointDtype() local |