Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dgemm_pattern.cc134 MatrixDescriptor rhs_matrix, MatrixDescriptor output_matrix, in CreateTfrtOps()
283 MatrixDescriptor rhs_matrix = make_descriptor( in GemmOpConversionRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc290 MatrixDescriptor rhs_matrix = make_descriptor( in RunGemm() local