Searched defs:b_array (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | concat_test.cc | 155 auto b_array = CreatePatternedMatrix(1, 1, /*offset=*/64.0); in XLA_TEST_F() local
170 auto b_array = CreatePatternedMatrix(1, 1, /*offset=*/64.0); in XLA_TEST_F() local
183 auto b_array = CreatePatternedMatrix(2, 5, /*offset=*/64.0); in XLA_TEST_F() local
194 auto b_array = CreatePatternedMatrix(2, 5, /*offset=*/64.0); in XLA_TEST_F() local
219 auto b_array = CreatePatternedMatrix(5, 2, /*offset=*/64.0); in XLA_TEST_F() local
254 Array3D<float> b_array({ in XLA_TEST_F() local
291 Array3D<float> b_array({ in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_batch_matmul_op.cc | 117 std::vector<const Scalar *> b_array; in Compute() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 2276 const port::ArraySlice<DeviceMemory<Eigen::half> *> &b_array, int ldb, in DoBlasGemmBatched()
2294 const port::ArraySlice<DeviceMemory<float> *> &b_array, int ldb, float beta, in DoBlasGemmBatched()
2310 const port::ArraySlice<DeviceMemory<double> *> &b_array, int ldb, in DoBlasGemmBatched()
2327 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &b_array, in DoBlasGemmBatched()
2345 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &b_array, in DoBlasGemmBatched()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1929 const port::ArraySlice<DeviceMemory<float> *> &b_array, int ldb, float beta, in DoBlasGemmBatched()
1946 const port::ArraySlice<DeviceMemory<double> *> &b_array, int ldb, in DoBlasGemmBatched()
1964 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &b_array, in DoBlasGemmBatched()
1978 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &b_array, in DoBlasGemmBatched()
|
| /external/python/cpython3/Python/ |
| D | hamt.c | 323 PyObject *b_array[1]; member
|