Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtopk_rewriter.cc140 const bool has_batch = data->shape().rank() == 2; in SortIsInTopK() local
208 const bool has_batch = data->shape().rank() == 2; in TransformToCustomCall() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2391 const bool has_batch = hlo->shape().tuple_shapes(0).dimensions_size() == 2; in HandleTopK() local
/external/mesa3d/src/mesa/main/
Dmtypes.h2027 bool has_batch; member