Searched defs:b_array (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | concat_test.cc | 158 auto b_array = CreatePatternedMatrix(1, 1, /*offset=*/64.0); in XLA_TEST_F() local
173 auto b_array = CreatePatternedMatrix(1, 1, /*offset=*/64.0); in XLA_TEST_F() local
186 auto b_array = CreatePatternedMatrix(2, 5, /*offset=*/64.0); in XLA_TEST_F() local
197 auto b_array = CreatePatternedMatrix(2, 5, /*offset=*/64.0); in XLA_TEST_F() local
222 auto b_array = CreatePatternedMatrix(5, 2, /*offset=*/64.0); in XLA_TEST_F() local
257 Array3D<float> b_array({ in XLA_TEST_F() local
294 Array3D<float> b_array({ in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_blas.cc | 1328 const DeviceMemorySlice<Eigen::half> &b_array, int ldb, float beta, in DoBlasInternalImpl()
1345 int lda, const DeviceMemorySlice<float> &b_array, int ldb, float beta, in DoBlasInternalImpl()
1361 const DeviceMemorySlice<double> &b_array, int ldb, double beta, in DoBlasInternalImpl()
1377 const DeviceMemorySlice<std::complex<float>> &b_array, int ldb, in DoBlasInternalImpl()
1394 const DeviceMemorySlice<std::complex<double>> &b_array, int ldb, in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 885 const absl::Span<DeviceMemory<float> *const> &b_array, int ldb, float beta, in DoBlasGemmBatched()
903 const absl::Span<DeviceMemory<double> *const> &b_array, int ldb, in DoBlasGemmBatched()
922 const absl::Span<DeviceMemory<std::complex<float>> *const> &b_array, in DoBlasGemmBatched()
942 const absl::Span<DeviceMemory<std::complex<double>> *const> &b_array, in DoBlasGemmBatched()
|
| /external/python/cpython3/Python/ |
| D | hamt.c | 324 PyObject *b_array[1]; member
|