| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | eval.cpp | 454 std::vector<DstType> castValues(const Dtype& src_dtype, const InterpValue& v) in castValues()
467 const Dtype& src_dtype, in doCastFromSrc()
499 Dtype src_dtype = src_value->dtype(); in visit() local
522 const Dtype& src_dtype, in bitcastValues()
534 const Dtype& src_dtype, in doBitCastFromSrc()
554 Dtype src_dtype = src_value->dtype(); in visit() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/ |
| H A D | cutlass_epilogue_gen.py | 200 def reduction(self, dtype, src_dtype, reduction_type, value): argument
343 def _op_to_dtype(self, a, dtype, src_dtype=None): argument
357 def reduction(self, dtype, src_dtype, reduction_type, value): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp.py | 204 src_dtype=None, argument
544 def to_dtype(x, dtype, src_dtype=None, use_compute_types=True): argument
587 def to_dtype_bitcast(x, dtype, src_dtype): argument
1441 def to_dtype(x, dtype, src_dtype=None, use_compute_dtypes=True): argument
1920 def reduction(self, dtype, src_dtype, reduction_type, value): argument
2195 def get_to_dtype_expr(self, src, dtype, src_dtype): argument
2198 def cache_dtype_convert(self, dst, dst_dtype, src, src_dtype): argument
2539 def reduction(self, dtype, src_dtype, reduction_type, value): argument
2961 def get_to_dtype_expr(self, src, dtype, src_dtype): argument
|
| H A D | triton_split_scan.py | 78 def reduction(self, dtype, src_dtype, reduction_type, value): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | python_tensor_converter.cc | 97 DataType src_dtype = DataTypeForTensor(src); in Convert() local
|
| /aosp_15_r20/external/pytorch/torch/utils/_sympy/ |
| H A D | value_ranges.py | 466 def to_dtype(a, dtype, src_dtype=None): argument
968 def reduction(self, name, dtype, src_dtype, reduction_type, index, value): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | xla_client_test.py | 803 def testConvertElementType(self, src_dtype, dst_dtype): argument
833 def testBitcastConvertType(self, src_dtype, dst_dtype): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | loop_body.py | 452 def reduction(self, dtype, src_dtype, reduction_type, value): argument
|
| H A D | utils.py | 1818 src_dtype, argument
|
| H A D | ir.py | 675 src_dtype, argument
898 def _unroll_reduction_fn(inner_fn, reduction_ranges, reduction_type, src_dtype): argument
8077 def reduction(self, dtype, src_dtype, reduction_type, value): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 4310 def helper(src_shape, dst_shape, src_dtype, dst_dtype): argument
6799 def helper(src_dtype, dst_dtype): argument
6810 def helper(src_dtype, dst_dtype): argument
|
| H A D | test_torch.py | 6349 def can_cast(src_dtype, dst_dtype): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_runtime.cc | 139 const auto src_dtype = R"JIT( in TEST() local
|