Searched defs:idx_dtype (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/ |
| D | test_mps.py | 7223 … def helper(shape, dim, index, source_shape, alpha, x_dtype=torch.float32, idx_dtype=torch.int32): argument
7329 def helper(shape, dim, index, idx_dtype=torch.int32): argument
7351 def helper(value, dim, index, idx_dtype=torch.int32): argument
7401 def helper(shape, dim, idx_shape, idx_dtype=torch.int64): argument
7456 def helper(shape, dim, idx_shape, src_shape, idx_dtype=torch.int64, do_add=True): argument
7523 def helper(idx_dtype=torch.int64, do_add=True): argument
7563 def helper(shape, dim, idx_shape, src_shape, idx_dtype=torch.int64, reduce_str="sum"): argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3590 DataType idx_dtype{DataType::DT_INT32}; in ConvertReduce() local
|