Searched refs:a_matrix (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gemm_functors.h | 101 typename tensorflow::TTypes<const T1>::Matrix a_matrix(a, m, k); in operator()
109 a_matrix.contract(b_matrix, dim_pair); in operator()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 698 auto& a_matrix = op_info.inputs(0); in CountMatMulOperations() local
717 MaybeGetMinimumShape(a_matrix.shape(), 2, found_unknown_shapes); in CountMatMulOperations()
841 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local
842 a_matrix->set_dtype(a_input.dtype()); in CountBatchMatMulOperations()
843 TensorShapeProto* a_matrix_shape = a_matrix->mutable_shape(); in CountBatchMatMulOperations()
|
| /external/eigen/Eigen/src/Eigenvalues/ |
| D | SelfAdjointEigenSolver.h | 400 ::compute(const EigenBase<InputType>& a_matrix, int options)
404 const InputType &matrix(a_matrix.derived());
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 2258 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; in DoBlasGemmBatchedInternal() local
2261 bool ok = DoBlasGemm(stream, transa, transb, m, n, k, alpha, a_matrix, in DoBlasGemmBatchedInternal()
2395 const auto *a_matrix = in DoBlasGemmStridedBatched() local
2404 CUDABlasTranspose(transb), m, n, k, &alpha, a_matrix, SE_CUDA_DATA_HALF, in DoBlasGemmStridedBatched()
|