Searched full:numtotensor (Results 1 – 20 of 20) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | peephole.cpp | 138 if (input_node->kind() == prim::NumToTensor) { in optimizeBlock()
141 " (x.NumToTensor() == x) is replaced with ", in optimizeBlock()
|
| H A D | erase_number_types.h | 14 // - prim::TensorToNum, aten::Float, aten::Int and prim::NumToTensor nodes
|
| H A D | erase_number_types.cpp | 55 case prim::NumToTensor: { in EraseNumberTypesOnBlock()
|
| H A D | remove_inplace_ops.cpp | 117 if ((shape_node->kind() == prim::NumToTensor) && in ImplicitCastForBinaryInplaceOps()
|
| H A D | shape_analysis.cpp | 609 case prim::NumToTensor: { in propagateNode()
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_dtype_analysis.py | 59 # create_traced_fn generates prim::NumToTensor nodes in graph (not supported yet)
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | interned_strings.h | 99 _(prim, NumToTensor) \
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/ |
| H A D | promoted_prim_ops.cpp | 240 mobile::prim_op_fn_register("prim::NumToTensor.Scalar", numToTensorScalar),
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | model_ops.yaml | 400 prim::NumToTensor.Scalar: 18
|
| H A D | coverage.yaml | 1080 prim::NumToTensor.Scalar: 15
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | converter.py | 884 # Converts prim::NumToTensor as aten.scalar_tensor.
885 # prim::NumToTensor IRs are currently triggered by:
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _shape_functions.py | 1241 "prim::NumToTensor.Scalar(Scalar a) -> Tensor", zero_dim_tensor
1243 add_shape_compute_mapping("prim::NumToTensor.bool(bool a) -> Tensor", zero_dim_tensor)
|
| /aosp_15_r20/external/pytorch/ |
| H A D | pt_ops.bzl | 194 "prim::NumToTensor.Scalar",
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | ops.cpp | 2819 prim::NumToTensor,
2823 torch::schema("prim::NumToTensor.Scalar(Scalar s) -> Tensor")) || in __anon11f46a8b9702()
2825 torch::schema("prim::NumToTensor.bool(bool a) -> Tensor"))) { in __anon11f46a8b9702()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/interpreter/ |
| H A D | code_impl.h | 829 case prim::NumToTensor: in emitNode()
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_runtime.cc | 3008 %3 : Tensor = prim::NumToTensor(%1) in TEST()
3022 %3 : Tensor = prim::NumToTensor(%1) in TEST()
3036 %3 : Tensor = prim::NumToTensor(%1) in TEST()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_prim_ops.cpp | 485 TORCH_SELECTIVE_SCHEMA("prim::NumToTensor.Scalar(Scalar a) -> Tensor"),
2412 TORCH_SELECTIVE_SCHEMA("prim::NumToTensor.bool(bool a) -> Tensor"),
|
| H A D | serialized_shape_function_registry.cpp | 3263 {"prim::NumToTensor.Scalar(Scalar a) -> Tensor", "zero_dim_tensor"}, in GetShapeFunctionMappings()
3264 {"prim::NumToTensor.bool(bool a) -> Tensor", "zero_dim_tensor"}, in GetShapeFunctionMappings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | ir.cpp | 1846 Node* result = create(prim::NumToTensor, {value}); in createNumToTensor()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | ir_emitter.cpp | 500 std::make_shared<CastValue>(TensorType::get(), prim::NumToTensor)}, in getSugaredVar()
|