| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_fused_batchnorm_test.py | 62 x_dtype, argument
143 x_dtype, argument
224 x_dtype, argument
287 x_dtype, argument
|
| H A D | nn_fused_batchnorm_deterministic_test.py | 50 def _genParams(self, data_format, x_dtype, large_batch): argument
|
| H A D | nn_fused_batchnorm_d9m_test.py | 48 def _genParams(self, data_format, x_dtype, large_batch): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| H A D | sparse_matmul_op_test.py | 46 x_dtype=dtypes.float32, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | TensorShape.cpp | 91 const auto x_dtype = qxs[0].get().scalar_type(); in quantized_cat_impl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_tensor_supported_values_test.py | 252 x_dtype=None, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | MKLDNNConversions.cpp | 293 ideep::data_type x_dtype, in get_conv_transpose_expected_weights_desc()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | type_spec_test.py | 58 def __init__(self, x_shape, x_dtype, y_shape, y_dtype, color="red"): argument
|
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/lib/ |
| H A D | test_function_base.py | 1148 def test_x_decreasing_unsigned(self, x_dtype): argument
1155 def test_x_signed_int_big_jump(self, x_dtype): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | math_grad.cc | 519 DataType x_dtype = x.type(); in PowGrad() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 7223 … def helper(shape, dim, index, source_shape, alpha, x_dtype=torch.float32, idx_dtype=torch.int32): argument
7813 def helper(shape, x_shape, y_shape, cond_dtype=torch.bool, x_dtype=torch.float): argument
|