Home
last modified time | relevance | path

Searched refs:input_image_size (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.cc190 const int64 input_image_size = in operator ()() local
209 const int64 in_base = b * input_image_size; in operator ()()
Ddepthwise_conv_grad_op.cc409 const int64 input_image_size = in operator ()() local
446 in_backprop + b * input_image_size); in operator ()()
869 const int64 input_image_size = in operator ()() local
886 input + b * input_image_size, input_buffer_data); in operator ()()
Dpooling_ops_common.h525 const int64 input_image_size =
532 in_flat.data() + start * input_image_size, params.depth,
Ddeep_conv2d.cc1114 const int64 input_image_size = args.in_rows * args.in_cols * in_depth; in operator ()() local
1121 const int64 in_base = b * input_image_size; in operator ()()