Searched defs:b_dtype (Results 1 – 6 of 6) sorted by relevance
| /external/executorch/kernels/portable/cpu/ |
| D | scalar_utils.h | 71 ScalarType b_dtype = get_scalar_dtype(b); in scalars_have_same_dtype() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops_test.py | 294 def testInt8MatMul2D(self, a_dtype, b_dtype): argument
304 def testMixPrecMatMul2D(self, b_dtype): argument
319 def testInt8BatchMatmul(self, a_dtype, b_dtype): argument
330 def testMixPrecBatchMatmul(self, b_dtype): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_matmul_op_test.py | 133 def _testGradients(self, tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype, delta, argument
|
| /external/pytorch/aten/src/ATen/cuda/tunable/ |
| D | GemmCommon.h | 318 ScalarType b_dtype; member
|
| /external/pytorch/test/ |
| D | test_fx_passes.py | 777 def pattern(a, a_dtype, a_scale, b, b_dtype, b_scale, out_scale): argument
|
| /external/pytorch/aten/src/ATen/test/ |
| D | vulkan_quantized_api_test.cpp | 2485 const c10::ScalarType b_dtype, in test_quantized_conv2d()
|