Searched full:scalarother (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/test/export/ |
| D | testing.py | 111 aten.where.ScalarOther,
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | symbolic_shape_registry_util.cpp | 93 … {"aten::where.ScalarOther(Tensor condition, Tensor self, Scalar other) -> Tensor", "broadcast"}, in get_tensorexpr_elementwise_set()
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesDecompositions.cpp | 311 OP_DECOMPOSE2(where, ScalarOther); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/test/mobile/model_test/ |
| D | model_ops.yaml | 389 aten::where.ScalarOther: 3
|
| D | coverage.yaml | 639 - aten::where.ScalarOther
1011 aten::where.ScalarOther: 4
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | lowerings.cpp | 1624 {"aten::where.ScalarOther(Tensor condition, Tensor self, Scalar other) -> (Tensor)", in nnc_lowerings_lazy_registration()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | native_functions.yaml | 6456 - func: where.ScalarOther(Tensor condition, Tensor self, Scalar other) -> Tensor
|