| /external/pytorch/torch/csrc/jit/passes/ |
| D | restore_mutation.cpp | 25 bool check_dtype = activation_type_promotion_mapping.at(node->kind()); in CanBeInplace() local
|
| /external/pytorch/test/onnx/ |
| D | test_pytorch_jit_onnx.py | 53 check_dtype = True variable in _TestJITIRToONNX
|
| D | onnx_test_common.py | 132 check_dtype = True variable in _TestONNXRuntime
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 136 def check_dtype(arg, dtype): function
|
| D | linear_operator_test_util.py | 94 def assertAC(self, x, y, check_dtype=False): argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | stateful_random_ops_test.py | 211 def check_dtype(self, dtype): member in StatefulRandomOpsTest
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_math_ops_test.py | 152 def match(self, actual, expected, msg='', check_dtype=True): argument
|
| D | np_array_ops_test.py | 1046 def match(self, actual, expected, msg=None, check_dtype=True): argument
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | pad_mm.py | 80 def check_dtype(a: Tensor, b: Tensor) -> bool: function
|
| /external/pytorch/torch/_numpy/testing/ |
| D | utils.py | 1230 check_dtype=False, argument
|
| /external/pytorch/torch/onnx/ |
| D | verification.py | 82 check_dtype: bool = True variable in VerificationOptions
|
| /external/executorch/backends/xnnpack/runtime/ |
| D | XNNCompiler.cpp | 646 auto check_dtype = [graph](uint32_t id, DataType dtype) -> bool { in isQP8() local
|
| /external/pytorch/torch/sparse/ |
| D | _triton_ops.py | 57 def check_dtype(f_name, t, dtype, *additional_dtypes): function
|
| /external/pytorch/test/torch_np/numpy_tests/linalg/ |
| D | test_linalg.py | 1329 def check_dtype(x, res): member in _TestNormBase
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_utils.py | 2608 …self, actual, expected, *, rtol_override=0.0, atol_override=0.0, check_dtype=None, **other_paramet… argument
|