Searched defs:input_image_size (Results 1 – 6 of 6) sorted by relevance
54 int64_t input_image_size = numel / channels; in cpu_max_unpool() local118 int64_t input_image_size = input_height * input_width; in cpu_max_unpool_channels_last() local191 int64_t input_image_size = numel / channels; in cpu_max_unpool_backward() local
44 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local
425 const int64_t input_image_size = in operator ()() local914 const int64_t input_image_size = in operator ()() local
198 const int64_t input_image_size = in operator ()() local
1132 const int64_t input_image_size = args.in_rows * args.in_cols * in_depth; in operator ()() local
4965 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local5464 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local