Searched refs:input_image_size (Results 1 – 6 of 6) sorted by relevance
39 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local47 const int input_offset = input_image_size * input_depth; in TransposeConvV2()77 rhs_params.cols = input_image_size; in TransposeConvV2()83 dst_params.cols = input_image_size; in TransposeConvV2()
195 const int64 input_image_size = in operator ()() local214 const int64 in_base = b * input_image_size; in operator ()()
421 const int64 input_image_size = in operator ()() local458 in_backprop + b * input_image_size); in operator ()()910 const int64 input_image_size = in operator ()() local927 input + b * input_image_size, input_buffer_data); in operator ()()
577 const int64 input_image_size =584 in_flat.data() + start * input_image_size, params.depth,
1114 const int64 input_image_size = args.in_rows * args.in_cols * in_depth; in operator ()() local1121 const int64 in_base = b * input_image_size; in operator ()()
5726 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local5734 const int input_offset = input_image_size * input_depth; in TransposeConvV2()5761 rhs_params.cols = input_image_size; in TransposeConvV2()5765 dst_params.cols = input_image_size; in TransposeConvV2()5952 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local5960 const int input_offset = input_image_size * input_depth; in TransposeConvV2()5989 rhs_params.cols = input_image_size; in TransposeConvV2()5995 dst_params.cols = input_image_size; in TransposeConvV2()