Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dtranspose_conv.h39 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.cc195 const int64 input_image_size = in operator ()() local
Ddepthwise_conv_grad_op.cc421 const int64 input_image_size = in operator ()() local
910 const int64 input_image_size = in operator ()() local
Ddeep_conv2d.cc1114 const int64 input_image_size = args.in_rows * args.in_cols * in_depth; in operator ()() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5726 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local
5952 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local