Home
last modified time | relevance | path

Searched defs:rhs_batch (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Djitrt_custom_calls.h41 llvm::ArrayRef<int64_t> rhs_batch; member
Djitrt_custom_calls.cc333 ArrayRef<int64_t> rhs_batch, in GetGemmConfig()
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc1198 const int64_t rhs_batch = in CanPropagate() local
1252 const int64_t rhs_batch = in CanPropagate() local