| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | cwise_ops_binary_test.py | 204 self._compareBoth(x, y + 0.1, np.floor_divide, math_ops.floordiv)
209 self._compareBoth(x, y + 0.1, np.floor_divide, _FLOORDIV)
271 self._compareBoth(x, y + 0.1, np.floor_divide, math_ops.floordiv)
276 self._compareBoth(x, y + 0.1, np.floor_divide, _FLOORDIV)
308 self._compareBoth(x, y, np.floor_divide, math_ops.floordiv)
312 self._compareBoth(x, y, np.floor_divide, _FLOORDIV)
330 self._compareBoth(x, y, np.floor_divide, math_ops.floordiv)
332 self._compareBoth(x, y, np.floor_divide, _FLOORDIV)
341 self._compareBoth(x, y, np.floor_divide, math_ops.floordiv)
347 self._compareBoth(x, y, np.floor_divide, _FLOORDIV)
[all …]
|
| H A D | basic_gpu_test.py | 59 self._compareGPU(x, y + 0.1, np.floor_divide, math_ops.floordiv)
139 np_out = np.floor_divide(x, y + 0.1)
196 self._compareGpu(x1, x2 + 0.1, np.floor_divide, math_ops.floordiv)
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_floor_divide_test.cpp | 36 // Destination for the floor_divide. in test_integer_floor_divide()
39 // floor_divide two tensors. in test_integer_floor_divide()
56 // Destination for the floor_divide. in test_floating_point_floor_divide()
59 // floor_divide two tensors. in test_floating_point_floor_divide()
78 // Destination for the floor_divide. in TEST_F()
81 // floor_divide two tensors. in TEST_F()
117 // floor_divide() doesn't handle Bool. in TEST_F()
143 // Destination for the floor_divide; matches the shape of one of the inputs. in TEST_F()
|
| H A D | test_case_gen.py | 742 x, y, (), (), torch_fn=torch.floor_divide
744 + make_test_cases_dynamic_shape(x, y, torch.floor_divide(x, y)),
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_floor_divide.cpp | 51 static constexpr const char op_name[] = "floor_divide.out"; in floor_divide_out()
65 return utils::floor_divide(val_a, val_b); in floor_divide_out()
|
| H A D | op_div.cpp | 140 value = utils::floor_divide(val_a, val_b); in div_out_mode()
263 value = utils::floor_divide(val_a, val_b); in div_scalar_mode_out()
|
| /aosp_15_r20/external/pytorch/test/typing/pass/ |
| H A D | math_ops.py | 148 # floor_divide
149 torch.floor_divide(torch.tensor([4.0, 3.0]), torch.tensor([2.0, 2.0]))
150 torch.floor_divide(torch.tensor([4.0, 3.0]), 1.4)
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | BinaryOp.cpp | 115 DEFINE_BINARY_OP_WITH_ALPHA_FN(floor_divide);
127 VK_REGISTER_OP(aten.div.Tensor_mode, floor_divide);
|
| /aosp_15_r20/external/executorch/backends/apple/mps/operators/ |
| H A D | binary_ops.py | 48 "aten.floor_divide.default",
75 exir_ops.edge.aten.floor_divide.default: MPSDiv,
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | math_util.h | 26 INT_T floor_divide(INT_T a, INT_T b) { in floor_divide() function
39 FLOAT_T floor_divide(FLOAT_T a, FLOAT_T b) { in floor_divide() function
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 241 - aten::floor_divide
242 - aten::floor_divide.Scalar
819 aten::floor_divide: 4
820 aten::floor_divide.Scalar: 7
|
| H A D | model_ops.yaml | 153 aten::floor_divide: 8
154 aten::floor_divide.Scalar: 37
|
| H A D | math_ops.py | 97 torch.floor_divide(torch.tensor([4.0, 3.0]), torch.tensor([2.0, 2.0])),
98 torch.floor_divide(torch.tensor([4.0, 3.0]), 1.4),
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_flex_decoding.py | 596 floor_divide = torch.ops.aten.floor_divide.default(arg3_1, add); arg3_1 = add = None
597 sub = torch.ops.aten.sub.Tensor(arg0_1, floor_divide); arg0_1 = floor_divide = None
|
| H A D | test_flex_attention.py | 606 floor_divide = torch.ops.aten.floor_divide.default(arg3_1, add); arg3_1 = add = None
607 sub = torch.ops.aten.sub.Tensor(arg0_1, floor_divide); arg0_1 = floor_divide = None
|
| /aosp_15_r20/external/pytorch/test/torch_np/ |
| H A D | test_binary_ufuncs.py | 61 np.floor_divide(0.5, 0.6),
62 floor_divide(0.5, 0.6),
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | BinaryOp.cpp | 588 self, other, std::optional<Scalar>(), VK_KERNEL(floor_divide)); in floor_divide_tensor()
625 TORCH_SELECTIVE_NAME("aten::floor_divide.Scalar"), in TORCH_LIBRARY_IMPL()
631 TORCH_SELECTIVE_NAME("aten::floor_divide"), in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/functorch/dim/ |
| H A D | op_properties.py | 147 torch.Tensor.floor_divide,
148 torch.floor_divide,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | NamedRegistrations.cpp | 195 m.impl("floor_divide", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
196 m.impl("floor_divide.Scalar", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
197 m.impl("floor_divide.out", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _pointwise_ops.py | 39 # "floor_divide", # floor_divide is deprecated
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BinaryOps.cpp | 46 #include <ATen/ops/floor_divide.h>
978 Tensor floor_divide(const Tensor& self, const Tensor& other) { in floor_divide() function
1510 Tensor floor_divide(const Tensor& self, const Scalar& other) { in floor_divide() function
1511 return at::floor_divide(self, wrapped_scalar_tensor(other)); in floor_divide()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesBinaryOps.cpp | 400 BINARY_POINTWISE(floor_divide); in TORCH_LIBRARY_IMPL()
401 UNARY_POINTWISE2(floor_divide, Scalar); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/tools/pyi/ |
| H A D | gen_pyi.py | 172 "floor_divide",
1004 for binop in ["true_divide", "floor_divide"]:
1232 for binop in ["true_divide", "floor_divide"]:
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _binary_ufuncs_impl.py | 20 floor_divide,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/glsl/ |
| H A D | shader_params.yaml | 44 - NAME: floor_divide
|