Searched defs:b_array (Results 1 – 3 of 3) 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
169 auto b_array = CreatePatternedMatrix(1, 1, /*offset=*/64.0); in XLA_TEST_F() local
182 auto b_array = CreatePatternedMatrix(2, 5, /*offset=*/64.0); in XLA_TEST_F() local
193 auto b_array = CreatePatternedMatrix(2, 5, /*offset=*/64.0); in XLA_TEST_F() local
218 auto b_array = CreatePatternedMatrix(5, 2, /*offset=*/64.0); in XLA_TEST_F() local
253 Array3D<float> b_array({ in XLA_TEST_F() local
288 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 | 2397 const port::ArraySlice<DeviceMemory<float> *> &b_array, int ldb, float beta, in DoBlasGemmBatched()
2413 const port::ArraySlice<DeviceMemory<double> *> &b_array, int ldb, in DoBlasGemmBatched()
2430 const port::ArraySlice<DeviceMemory<std::complex<float>> *> &b_array, in DoBlasGemmBatched()
2448 const port::ArraySlice<DeviceMemory<std::complex<double>> *> &b_array, in DoBlasGemmBatched()
|