Searched refs:a_index (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | masked_matmul_ops.cc | 120 int64 a_index = internal::SubtleMustCopy(indices_mat(i, 0)); in Compute() local
121 CHECK(FastBoundsCheck(a_index, a_dim_0)) in Compute()
122 << "In mask_indices[" << i << ", :], the row index " << a_index in Compute()
124 return a_index; in Compute()
178 const int64 a_index = get_a_index(p); in Compute() local
180 prod_values(p) = get_dot_product(a_index, b_index); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reference_gemm.h | 75 const size_t a_index = ((i * a_i_stride) + (l * a_l_stride)); in ReferenceGemm() local
76 const int32 a_value = static_cast<int32>(a[a_index]) - offset_a; in ReferenceGemm()
|
| D | gemm_functors.h | 78 const size_t a_index = ((i * a_i_stride) + (l * a_l_stride)); in operator() local
79 const T1 a_value = a[a_index]; in operator()
|
| /external/mesa3d/src/gallium/drivers/r300/compiler/ |
| D | radeon_variable.c | 240 unsigned int a_index, b_index; in readers_intersect() local
241 for (a_index = 0; a_index < a->ReaderCount; a_index++) { in readers_intersect()
242 struct rc_reader reader_a = a->Readers[a_index]; in readers_intersect()
|
| /external/eigen/Eigen/src/Core/ |
| D | Diagonal.h | 73 …licit inline Diagonal(MatrixType& matrix, Index a_index = DiagIndex) : m_matrix(matrix), m_index(a…
|
| /external/mesa3d/src/mesa/drivers/dri/i965/ |
| D | brw_performance_query.c | 581 accumulate_uint40(int a_index, in accumulate_uint40() argument
588 uint64_t high0 = (uint64_t)(high_bytes0[a_index]) << 32; in accumulate_uint40()
589 uint64_t high1 = (uint64_t)(high_bytes1[a_index]) << 32; in accumulate_uint40()
590 uint64_t value0 = report0[a_index + 4] | high0; in accumulate_uint40()
591 uint64_t value1 = report1[a_index + 4] | high1; in accumulate_uint40()
|
| /external/libvpx/libvpx/vpx_dsp/arm/ |
| D | fdct32x32_neon.c | 539 #define ADDW_S16_S32(a, a_index, b, b_index, c, c_index) \ argument
541 c##_lo[c_index] = vaddw_s16(a##_lo[a_index], vget_low_s16(b[b_index])); \
542 c##_hi[c_index] = vaddw_s16(a##_hi[a_index], vget_high_s16(b[b_index])); \
545 #define SUBW_S16_S32(a, a_index, b, b_index, temp, temp_index, c, c_index) \ argument
547 temp##_lo[temp_index] = vmovl_s16(vget_low_s16(a[a_index])); \
548 temp##_hi[temp_index] = vmovl_s16(vget_high_s16(a[a_index])); \
|
| /external/gemmlowp/test/ |
| D | test.cc | 84 const int a_index = i * a_i_stride + l * a_l_stride; in ReferenceEightBitIntGemm() local
85 const std::uint8_t a_as_byte = a[a_index]; in ReferenceEightBitIntGemm()
|