Home
last modified time | relevance | path

Searched full:tensor_mode (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/csrc/jit/operator_upgraders/
H A Dversion_map.cpp38 {"aten::div.Tensor_mode",
41 "aten::div.Tensor_mode(Tensor self, Tensor other, *, str? rounding_mode) -> Tensor"}}},
62 {"aten::div_.Tensor_mode",
65 … "aten::div_.Tensor_mode(Tensor(a!) self, Tensor other, *, str? rounding_mode) -> Tensor(a!)"}}},
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/
H A Dupgrader_mobile.cpp37 {std::string("aten::div.Tensor_mode"), in getOperatorVersionMapForMobile()
61 {std::string("aten::div_.Tensor_mode"), in getOperatorVersionMapForMobile()
200 OperatorString({"aten::div", "Tensor_mode", 3}), in getUpgraderBytecodeList()
219 OperatorString({"aten::div", "Tensor_mode", 3}), in getUpgraderBytecodeList()
326 OperatorString({"aten::div_", "Tensor_mode", 3}), in getUpgraderBytecodeList()
345 OperatorString({"aten::div_", "Tensor_mode", 3}), in getUpgraderBytecodeList()
/aosp_15_r20/external/executorch/backends/apple/mps/operators/
H A Dbinary_ops.py43 "aten.div.Tensor_mode",
70 exir_ops.edge.aten.div.Tensor_mode: MPSDiv,
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_aten_core_operators.expect191 aten::div.Tensor_mode
197 aten::div_.Tensor_mode
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesDecompositions.cpp357 OP_DECOMPOSE2(divide, Tensor_mode); in TORCH_LIBRARY_IMPL()
358 OP_DECOMPOSE2(divide_, Tensor_mode); in TORCH_LIBRARY_IMPL()
H A DBatchRulesBinaryOps.cpp398 BINARY_SCALAR_2(div, Tensor_mode, Scalar_mode); in TORCH_LIBRARY_IMPL()
474 …VMAP_SUPPORT2(div_, Tensor_mode, SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorInplaceModeT… in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/mobile/
H A Dtest_upgrader_bytecode_table_example.cpp149 OperatorString({"aten::div", "Tensor_mode", 3}), in getUpgraderBytecodeList()
237 OperatorString({"aten::div_", "Tensor_mode", 3}), in getUpgraderBytecodeList()
/aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/
H A D_pointwise_ops.py174 aten.div.Tensor_mode,
178 aten.div_.Tensor_mode,
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Donnxfunction_dispatcher.py447 Real example: `aten::div` vs `aten::div.Tensor_mode`.
481 Real example: `aten::div` vs `aten::div.Tensor_mode`.
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dts_native_functions.yaml37 - div.Tensor_mode
H A DBinaryOps.cpp176 TORCH_META_FUNC2(div, Tensor_mode) (const Tensor& self, const Tensor& other, std::optional<c10::str… in TORCH_META_FUNC2() argument
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DBinaryOp.cpp127 VK_REGISTER_OP(aten.div.Tensor_mode, floor_divide);
/aosp_15_r20/external/executorch/backends/vulkan/partitioner/
H A Dsupported_ops.py56 exir_ops.edge.aten.div.Tensor_mode,
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml187 - aten::div.Tensor_mode
792 aten::div.Tensor_mode: 7
H A Dmodel_ops.yaml120 aten::div.Tensor_mode: 12
/aosp_15_r20/external/pytorch/test/export/
H A Dtesting.py15 aten.divide.Tensor_mode,
/aosp_15_r20/external/executorch/backends/vulkan/
H A Dop_registry.py250 exir_ops.edge.aten.div.Tensor_mode,
/aosp_15_r20/external/pytorch/torch/_export/
H A Dconverter.py918 # converts aten::div.Tensor_mode(x, tensor_constant)
920 if schema.overload_name == "Tensor_mode":
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_flex_decoding.py610 div = torch.ops.aten.div.Tensor_mode(arg3_1, add, rounding_mode = 'floor'); arg3_1 = add = None
H A Dtest_flex_attention.py620 div = torch.ops.aten.div.Tensor_mode(arg3_1, add, rounding_mode = 'floor'); arg3_1 = add = None
/aosp_15_r20/external/executorch/exir/dialects/edge/op/
H A Dsample_input.py412 "div.Tensor_mode": { # (Tensor self, Tensor other, *, str? rounding_mode) -> Tensor
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/passes/
H A Dtype_promotion.py1063 # torch.ops.aten.div.Tensor_mode applies different type promotion rules
/aosp_15_r20/external/executorch/exir/dialects/edge/
H A Dedge.yaml3061 - func: aten::div.Tensor_mode
3063 inherits: aten::div.Tensor_mode
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchingRegistrations.cpp1214 …m.impl("div.Tensor_mode", binary_pointwise_batching_rule<Binop, at::div, std::optional<c10::string… in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/
H A Dtest_meta.py366 aten.div.Tensor_mode,

12