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/stream_executor/cuda/ |
| D | cuda_blas.cc | 2448 const port::ArraySlice<DeviceMemory<Eigen::half> *> &b_array, int ldb, in DoBlasInternalImpl()
2466 const port::ArraySlice<DeviceMemory<float> *> &b_array, int ldb, float beta, in DoBlasInternalImpl()
2482 const port::ArraySlice<DeviceMemory<double> *> &b_array, int ldb, in DoBlasInternalImpl()
2499 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &b_array, in DoBlasInternalImpl()
2517 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &b_array, in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1844 const port::ArraySlice<DeviceMemory<float> *> &b_array, int ldb, float beta, in DoBlasGemmBatched()
1862 const port::ArraySlice<DeviceMemory<double> *> &b_array, int ldb, in DoBlasGemmBatched()
1881 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &b_array, in DoBlasGemmBatched()
1901 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &b_array, in DoBlasGemmBatched()
|
| /external/python/cpython3/Python/ |
| D | hamt.c | 324 PyObject *b_array[1]; member
|