Searched defs:a_dtype (Results 1 – 5 of 5) sorted by relevance
| /external/executorch/kernels/portable/cpu/ |
| D | scalar_utils.h | 70 ScalarType a_dtype = get_scalar_dtype(a); in scalars_have_same_dtype() local
|
| /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/tensorflow/tensorflow/python/ops/ |
| D | math_ops_test.py | 294 def testInt8MatMul2D(self, a_dtype, b_dtype): argument
319 def testInt8BatchMatmul(self, a_dtype, b_dtype): argument
|
| /external/pytorch/aten/src/ATen/cuda/tunable/ |
| D | GemmCommon.h | 314 ScalarType a_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
|