Home
last modified time | relevance | path

Searched refs:a_array (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconcat_test.cc154 auto a_array = CreatePatternedMatrix(1, 1); in XLA_TEST_F() local
156 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F()
169 auto a_array = CreatePatternedMatrix(1, 1); in XLA_TEST_F() local
171 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F()
193 auto a_array = CreatePatternedMatrix(2, 3); in XLA_TEST_F() local
195 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F()
208 auto a_array = CreatePatternedMatrix(3, 2); in XLA_TEST_F() local
209 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F()
213 ComputeAndCompareR2<float>(&builder, *a_array, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
218 auto a_array = CreatePatternedMatrix(3, 2); in XLA_TEST_F() local
[all …]
/external/python/cpython3/Python/
Dhamt.c316 PyHamtNode *a_array[HAMT_ARRAY_NODE_SIZE]; member
383 if (node->a_array[i] != NULL) { in _hamt_node_array_validate()
865 new_node->a_array[jdx] = hamt_node_assoc( in hamt_node_bitmap_assoc()
867 if (new_node->a_array[jdx] == NULL) { in hamt_node_bitmap_assoc()
879 assert(new_node->a_array[i] == NULL); in hamt_node_bitmap_assoc()
882 new_node->a_array[i] = in hamt_node_bitmap_assoc()
884 Py_INCREF(new_node->a_array[i]); in hamt_node_bitmap_assoc()
892 new_node->a_array[i] = hamt_node_assoc( in hamt_node_bitmap_assoc()
899 if (new_node->a_array[i] == NULL) { in hamt_node_bitmap_assoc()
1654 node->a_array[i] = NULL; in hamt_node_array_new()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dmkl_batch_matmul_op.cc150 std::vector<const Scalar *> a_array; in Compute() local
153 a_array.reserve(batch_size); in Compute()
159 a_array.push_back(&lhs_reshaped(i, 0, 0)); in Compute()
170 a_array.push_back(&lhs_reshaped(a_batch_indices[i], 0, 0)); in Compute()
177 &k_array[0], &a_array[0], &lda_array[0], &b_array[0], in Compute()
/external/eigen/test/
Dmapstride.cpp23 Scalar* a_array = internal::aligned_new<Scalar>(arraysize+1); in map_class_vector() local
24 Scalar* array = a_array; in map_class_vector()
26 …array = (Scalar*)(internal::IntPtr(a_array) + (internal::packet_traits<Scalar>::AlignedOnScalar?si… in map_class_vector()
48 internal::aligned_delete(a_array, arraysize+1); in map_class_vector()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.h113 const port::ArraySlice<DeviceMemory<T> *> &a_array, int lda,
Dcuda_blas.cc2275 const port::ArraySlice<DeviceMemory<Eigen::half> *> &a_array, int lda, in DoBlasGemmBatched() argument
2282 cublasSgemmBatched, stream, transa, transb, m, n, k, alpha, a_array, lda, in DoBlasGemmBatched()
2293 const port::ArraySlice<DeviceMemory<float> *> &a_array, int lda, in DoBlasGemmBatched() argument
2298 cublasSgemmBatched, stream, transa, transb, m, n, k, alpha, a_array, lda, in DoBlasGemmBatched()
2309 const port::ArraySlice<DeviceMemory<double> *> &a_array, int lda, in DoBlasGemmBatched() argument
2314 cublasDgemmBatched, stream, transa, transb, m, n, k, alpha, a_array, lda, in DoBlasGemmBatched()
2325 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &a_array, in DoBlasGemmBatched() argument
2332 cublasCgemmBatched, stream, transa, transb, m, n, k, alpha, a_array, lda, in DoBlasGemmBatched()
2343 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &a_array, in DoBlasGemmBatched() argument
2350 cublasZgemmBatched, stream, transa, transb, m, n, k, alpha, a_array, lda, in DoBlasGemmBatched()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc1995 const port::ArraySlice<DeviceMemory<float> *> &a_array, int lda, in DoBlasGemmBatched() argument
2001 alpha, a_array, lda, b_array, ldb, beta, c_array, ldc, batch_count, in DoBlasGemmBatched()
2012 const port::ArraySlice<DeviceMemory<double> *> &a_array, int lda, in DoBlasGemmBatched() argument
2018 alpha, a_array, lda, b_array, ldb, beta, c_array, ldc, batch_count, in DoBlasGemmBatched()
2029 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &a_array, in DoBlasGemmBatched() argument
2043 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &a_array, in DoBlasGemmBatched() argument