Home
last modified time | relevance | path

Searched defs:lhs_matrix (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dgemm_pattern.cc133 int64_t batch_size, mlir::Type element_type, MatrixDescriptor lhs_matrix, in CreateTfrtOps()
280 MatrixDescriptor lhs_matrix = make_descriptor( in GemmOpConversionRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc287 MatrixDescriptor lhs_matrix = make_descriptor( in RunGemm() local