Home
last modified time | relevance | path

Searched defs:rhs_stride (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbackend_configs.proto64 int64 rhs_stride = 11; field
Dgemm_rewriter.cc75 int64 rhs_stride = rhs->shape().dimensions(lhs_batch_dims_size) * in HandleDot() local
/external/ruy/example/
Dparametrized_example.cc53 int rhs_stride; member
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dgemm_pattern.cc179 auto rhs_stride = builder.create<tfrt::compiler::ConstantI64Op>( in CreateTfrtOps() local