Home
last modified time | relevance | path

Searched defs:output_batch_size (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_batch_nd.cc80 const int output_batch_size = in ResizeOutputTensor() local
Dbatch_to_space_nd.cc78 const int output_batch_size = in ResizeOutputTensor() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h3096 const int output_batch_size = output_shape.Dims(0); in SpaceToBatchND() local
3149 const int output_batch_size = output_shape.Dims(0); in BatchToSpaceND() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5795 const int output_batch_size = output_shape.Dims(0); in BatchToSpaceND() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h6123 const int output_batch_size = output_shape.Dims(0); in BatchToSpaceND() local