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 | 2517 const port::ArraySlice<DeviceMemory<Eigen::half> *> &b_array, int ldb, in DoBlasInternalImpl()
2535 const port::ArraySlice<DeviceMemory<float> *> &b_array, int ldb, float beta, in DoBlasInternalImpl()
2551 const port::ArraySlice<DeviceMemory<double> *> &b_array, int ldb, in DoBlasInternalImpl()
2568 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &b_array, in DoBlasInternalImpl()
2586 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &b_array, in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 2159 const port::ArraySlice<DeviceMemory<float> *> &b_array, int ldb, float beta, in DoBlasGemmBatched()
2177 const port::ArraySlice<DeviceMemory<double> *> &b_array, int ldb, in DoBlasGemmBatched()
2196 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &b_array, in DoBlasGemmBatched()
2216 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &b_array, in DoBlasGemmBatched()
|
| /external/python/cpython3/Python/ |
| D | hamt.c | 323 PyObject *b_array[1]; member
|