Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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
/aosp_15_r20/out/soong/.intermediates/external/perfetto/perfetto_protos_perfetto_trace_processor_zero_gen_headers/gen/external/perfetto/protos/perfetto/trace_processor/
Dtrace_processor.pbzero.h1149 bool has_batch() const { return at<3>().valid(); } in has_batch() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc2391 const bool has_batch = hlo->shape().tuple_shapes(0).dimensions_size() == 2; in HandleTopK() local
/aosp_15_r20/external/mesa3d/src/mesa/main/
H A Dmtypes.h2017 bool has_batch; member