Home
last modified time | relevance | path

Searched refs:a_array (Results 1 – 9 of 9) 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.c315 PyHamtNode *a_array[HAMT_ARRAY_NODE_SIZE]; member
381 if (node->a_array[i] != NULL) { in _hamt_node_array_validate()
863 new_node->a_array[jdx] = hamt_node_assoc( in hamt_node_bitmap_assoc()
865 if (new_node->a_array[jdx] == NULL) { in hamt_node_bitmap_assoc()
877 assert(new_node->a_array[i] == NULL); in hamt_node_bitmap_assoc()
880 new_node->a_array[i] = in hamt_node_bitmap_assoc()
882 Py_INCREF(new_node->a_array[i]); in hamt_node_bitmap_assoc()
890 new_node->a_array[i] = hamt_node_assoc( in hamt_node_bitmap_assoc()
897 if (new_node->a_array[i] == NULL) { in hamt_node_bitmap_assoc()
1652 node->a_array[i] = NULL; in hamt_node_array_new()
[all …]
/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/core/kernels/
Dmkl_batch_matmul_op.cc116 std::vector<const Scalar *> a_array; in Compute() local
119 a_array.reserve(batch_size); in Compute()
123 a_array.push_back(&lhs_reshaped(i, 0, 0)); in Compute()
129 &k_array[0], &a_array[0], &lda_array[0], &b_array[0], in Compute()
/external/v8/benchmarks/
Dcrypto.js289 var a_array = a.array;
296 while(--i >= 0) if((r=this_array[i]-a_array[i]) != 0) return r;
381 var a_array = a.array;
384 c += this_array[i]-a_array[i];
400 c -= a_array[i];
815 var a_array = a.array;
818 for(i = 0; i < m; ++i) r_array[i] = op(this_array[i],a_array[i]);
826 for(i = m; i < a.t; ++i) r_array[i] = op(f,a_array[i]);
937 var a_array = a.array;
941 c += this_array[i]+a_array[i];
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.h106 const port::ArraySlice<DeviceMemory<T> *> &a_array, int lda,
Drocm_blas.cc1928 const port::ArraySlice<DeviceMemory<float> *> &a_array, int lda, in DoBlasGemmBatched() argument
1934 alpha, a_array, lda, b_array, ldb, beta, c_array, ldc, batch_count, in DoBlasGemmBatched()
1945 const port::ArraySlice<DeviceMemory<double> *> &a_array, int lda, in DoBlasGemmBatched() argument
1951 alpha, a_array, lda, b_array, ldb, beta, c_array, ldc, batch_count, in DoBlasGemmBatched()
1962 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &a_array, in DoBlasGemmBatched() argument
1976 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &a_array, in DoBlasGemmBatched() argument
/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()