Searched defs:compute_mode (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_cdist_forward_test.cpp | 29 optional<int64_t> compute_mode, in op_cdist_forward_out()
54 optional<int64_t> compute_mode = optional<int64_t>(); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Distance.cpp | 81 …ist_impl(const Tensor& x1, const Tensor& x2, const double p, std::optional<int64_t> compute_mode) { in cdist_impl()
150 …or cdist(const Tensor& x1, const Tensor& x2, const double p, std::optional<int64_t> compute_mode) { in cdist()
178 …_forward(const Tensor& x1, const Tensor& x2, const double p, std::optional<int64_t> compute_mode) { in _cdist_forward()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | distance_util.cpp | 35 optional<int64_t> compute_mode, in check_cdist_args()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_cdist_forward.cpp | 123 optional<int64_t> compute_mode, in _cdist_forward_out()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.proto | 162 DataType compute_mode = 4; field
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/r600/ |
| H A D | eg_debug.c | 132 const char *compute_mode = (ib[0] & 0x2) ? "(C)" : ""; in ac_parse_packet3() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | TensorModeKernel.cuh | 200 __global__ void compute_mode( in compute_mode() function
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | functional.py | 1431 def cdist(x1, x2, p=2.0, compute_mode="use_mm_for_euclid_dist_if_necessary"): argument
|
| H A D | _meta_registrations.py | 3290 def meta_cdist_forward(x1, x2, p, compute_mode): argument
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.pb.h | 1885 inline ::stream_executor::dnn::DataType ConvolutionDescriptorProto::compute_mode() const { in compute_mode() function
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 6189 compute_mode="use_mm_for_euclid_dist_if_necessary", argument
|