Searched refs:b_matrix (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gemm_functors.h | 102 typename tensorflow::TTypes<const T2>::Matrix b_matrix(b, k, n); in operator()
109 a_matrix.contract(b_matrix, dim_pair); in operator()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 699 auto& b_matrix = op_info.inputs(1); in CountMatMulOperations() local
719 MaybeGetMinimumShape(b_matrix.shape(), 2, found_unknown_shapes); in CountMatMulOperations()
849 OpInfo::TensorProperties* b_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local
850 b_matrix->set_dtype(b_input.dtype()); in CountBatchMatMulOperations()
851 TensorShapeProto* b_matrix_shape = b_matrix->mutable_shape(); in CountBatchMatMulOperations()
1246 auto b_matrix = op_info.inputs(3); in PredictSparseTensorDenseMatMul() local
1248 MaybeGetMinimumShape(b_matrix.shape(), 2, &found_unknown_shapes); in PredictSparseTensorDenseMatMul()
1262 num_elems_in_a * n_dim * DataTypeSize(BaseType(b_matrix.dtype())); in PredictSparseTensorDenseMatMul()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 2259 const DeviceMemory<T> &b_matrix = *b_ptrs_to_wrappers[b]; in DoBlasGemmBatchedInternal() local
2262 lda, b_matrix, ldb, beta, c_matrix, ldc); in DoBlasGemmBatchedInternal()
2397 const auto *b_matrix = in DoBlasGemmStridedBatched() local
2405 lda, b_matrix, SE_CUDA_DATA_HALF, ldb, &beta, c_matrix, in DoBlasGemmStridedBatched()
|