Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.h39 llvm::ArrayRef<int64_t> lhs_batch; member
H A Djitrt_custom_calls.cc331 ArrayRef<int64_t> lhs_batch, in GetGemmConfig()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dspace_to_batch_converter.cc1213 const int64_t lhs_batch = in CanPropagate() local
1242 const int64_t lhs_batch = activations->shape().dimensions( in CanPropagate() local