| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_llvm.cpp | 346 std::vector<int32_t> a_buffer = {42}; in TEST() local
361 std::vector<int32_t> a_buffer = {42}; in TEST() local
490 std::vector<int32_t> a_buffer = {1, 1, 1, 1}; in TEST() local
633 std::vector<int32_t> a_buffer(N, 42); in TEST() local
672 std::vector<int32_t> a_buffer(N, 41); in TEST() local
697 std::vector<float> a_buffer(N, 41); in TEST() local
721 std::vector<float> a_buffer(N, 10.0f); in TEST() local
747 std::vector<float> a_buffer(N, expf(3.0f) - 1); in TEST() local
774 std::vector<int> a_buffer(N, 41); in TEST() local
800 std::vector<int> a_buffer(N, 41); in TEST() local
[all …]
|
| H A D | test_aten.cpp | 947 std::vector<int> a_buffer(N, 1); in TEST() local
972 std::vector<int> a_buffer(N, 5); in TEST() local
997 std::vector<int> a_buffer(N, 6); in TEST() local
1022 std::vector<int> a_buffer(N, 5); in TEST() local
1047 std::vector<int> a_buffer(N, 5); in TEST() local
|
| H A D | test_expr.cpp | 334 std::vector<int> a_buffer(N, 1); in TEST() local
372 std::vector<int> a_buffer(N, 1); in TEST() local
409 std::vector<double> a_buffer(N, -10.0); in TEST() local
|
| /aosp_15_r20/frameworks/av/media/libstagefright/tests/fuzzers/ |
| H A D | SkipCutBufferFuzzer.cpp | 41 sp<ABuffer> a_buffer(new ABuffer(buf_size)); in LLVMFuzzerTestOneInput() local
55 sp<ABuffer> a_buffer(new ABuffer(buf_size)); in LLVMFuzzerTestOneInput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | cholesky_thunk.cc | 108 BufferAllocation::Slice a_buffer, in CholeskyThunk()
131 se::DeviceMemoryBase a_buffer = in ExecuteOnStream() local
|
| H A D | matmul_utils.cc | 687 Status MatmulPlan::DoMatmul(se::Stream* stream, se::DeviceMemoryBase a_buffer, in DoMatmul()
718 se::Stream* stream, se::DeviceMemoryBase a_buffer, in ExecuteOnStream()
|
| H A D | cublas_lt_matmul_thunk.cc | 32 BufferAllocation::Slice a_buffer, BufferAllocation::Slice b_buffer, in CublasLtMatmulThunk()
|
| H A D | cholesky_thunk.h | 73 se::DeviceMemoryBase a_buffer; member
|
| H A D | triangular_solve_thunk.cc | 37 const BufferAllocation::Slice& a_buffer, in TriangularSolveThunk()
|
| H A D | jitrt_custom_calls.cc | 1319 se::DeviceMemoryBase a_buffer = GetDeviceAddress(a); in operator ()() local
|
| /aosp_15_r20/external/executorch/kernels/optimized/vec/ |
| H A D | vec_base.h | 532 int_vector a_buffer = _mm256_load_si256(reinterpret_cast<const int_vector*>((const T*)a)); in bitwise_binary_op() local
535 int_vector a_buffer = _mm512_load_si512(reinterpret_cast<const int_vector*>((const T*)a)); in bitwise_binary_op() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cpu/vec/ |
| H A D | vec_base.h | 774 int_vector a_buffer = _mm256_load_si256(reinterpret_cast<const int_vector*>((const T*)a)); in bitwise_binary_op() local
777 int_vector a_buffer = _mm512_load_si512(reinterpret_cast<const int_vector*>((const T*)a)); in bitwise_binary_op() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebraUtils.h | 218 Tensor a_buffer, a_was_accessed, a_buffer_3d; in batch_iterator_with_broadcasting() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A D | vulkan_compute_api_test.cpp | 284 vTensor a_buffer = vTensor( in TEST_F() local
|