Searched full:prim_dtype (Results 1 – 12 of 12) sorted by relevance
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | symbolic_shape_registry_util.cpp | 30 …{"aten::to.prim_dtype(Tensor(a) self, int? dtype=None, bool non_blocking=False, bool copy=False) -… in get_tensorexpr_elementwise_set()
|
| D | register_prim_ops.cpp | 1269 …"aten::to.prim_dtype(Tensor(a) self, int? dtype=None, bool non_blocking=False, bool copy=False) ->…
|
| /external/pytorch/test/mobile/model_test/ |
| D | model_ops.yaml | 366 aten::to.prim_dtype: 89
|
| D | coverage.yaml | 1073 aten::to.prim_dtype: 38
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | native_ops.cpp | 526 …"aten::to.prim_dtype(Tensor(a) self, int? dtype, bool non_blocking=False, bool copy=False) -> Tens… in __anon69571ad13102()
540 "dytpe cannot be None when copy is True for aten::to.prim_dtype"); in __anon69571ad13102()
1324 prim_dtype,
|
| D | passes.cpp | 412 …"static_runtime::to_maybe_copy_out.prim_dtype(Tensor self, int? dtype=None, bool non_blocking=Fals… in TORCH_LIBRARY_FRAGMENT()
421 …"static_runtime::to_copy.prim_dtype(Tensor self, int? dtype=None, bool non_blocking=False, bool co… in TORCH_LIBRARY_FRAGMENT()
592 …"aten::to.prim_dtype(Tensor(a) self, int? dtype=None, bool non_blocking=False, bool copy=False) ->… in ReplaceWithMaybeCopy()
|
| D | ops.cpp | 1594 …"static_runtime::to_maybe_copy_out.prim_dtype(Tensor self, int? dtype=None, bool non_blocking=Fals… in __anon5bf8650b4202()
1640 …"static_runtime::to_copy.prim_dtype(Tensor self, int? dtype=None, bool non_blocking=False, bool co… in __anon5bf8650b4302()
|
| /external/pytorch/ |
| D | pt_ops.bzl | 356 "aten::to.prim_dtype",
|
| /external/pytorch/torch/_export/ |
| D | converter.py | 959 # special handle aten.to.dtype and aten.to.prim_dtype followed by inplace_mutation_op
964 if target == torch.ops.aten.to.dtype or target == torch.ops.aten.to.prim_dtype:
|
| /external/pytorch/docs/source/ |
| D | conf.py | 1638 "prim_dtype",
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | lowerings.cpp | 1579 …"aten::to.prim_dtype(Tensor(a) self, int? dtype=None, bool non_blocking=False, bool copy=False) ->… in nnc_lowerings_lazy_registration()
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset9.py | 207 "prim_dtype",
6346 def prim_dtype(g: jit_utils.GraphContext, self): function
|