Home
last modified time | relevance | path

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

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