Home
last modified time | relevance | path

Searched defs:out_batch (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfractional_avg_pool_op.cc244 const int64 out_batch = out_backprop.dim_size(0); in Compute() local
Dmkl_conv_ops.h372 int out_batch = GetTensorDim(input_shape, data_format_, 'N'); variable
Dconv_ops.cc667 const int64 out_batch = GetTensorDim(*output, data_format, 'N'); in operator ()() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h3162 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5808 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h6136 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local