Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Ddepthwiseconv_multithread.h145 const int output_batches = output_shape.Dims(0); in DepthwiseConv() local
H A Dlegacy_optimized_ops.h450 const int output_batches = output_shape.Dims(0); variable
552 const int output_batches = output_shape.Dims(0); variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Ddepthwise_conv_hybrid.h407 const int output_batches = output_shape.Dims(0); in DepthwiseConvHybridPerChannel() local