Searched refs:spatial_dim_index (Results 1 – 4 of 4) sorted by relevance
973 for (int64_t spatial_dim_index = 0; in RewriteDynamicConvolutionInputGrad() local974 spatial_dim_index < dnums.input_spatial_dimensions_size(); in RewriteDynamicConvolutionInputGrad()975 ++spatial_dim_index) { in RewriteDynamicConvolutionInputGrad()977 dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad()992 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad()999 padding_before[spatial_dim_index] = dynamic_window_dims.padding_before; in RewriteDynamicConvolutionInputGrad()1043 for (int64_t spatial_dim_index = 0; in RewriteDynamicConvolutionForward() local1044 spatial_dim_index < dnums.input_spatial_dimensions_size(); in RewriteDynamicConvolutionForward()1045 ++spatial_dim_index) { in RewriteDynamicConvolutionForward()1047 dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionForward()[all …]
752 for (int64_t spatial_dim_index = 0; in HandleDynamicConvolutionForward() local753 spatial_dim_index < dimension_numbers.input_spatial_dimensions_size(); in HandleDynamicConvolutionForward()754 ++spatial_dim_index) { in HandleDynamicConvolutionForward()756 dimension_numbers.input_spatial_dimensions(spatial_dim_index); in HandleDynamicConvolutionForward()758 dimension_numbers.output_spatial_dimensions(spatial_dim_index); in HandleDynamicConvolutionForward()761 WindowDimension window_dim = hlo->window().dimensions(spatial_dim_index); in HandleDynamicConvolutionForward()
532 for (int spatial_dim_index = 0, end = spatial_dims.size(); in DimensionsFromShape() local533 spatial_dim_index < end; ++spatial_dim_index) { in DimensionsFromShape()534 spatial_dims[spatial_dim_index] = context->Dim( in DimensionsFromShape()535 shape, GetTensorSpatialDimIndex(rank, format, spatial_dim_index)); in DimensionsFromShape()561 for (int spatial_dim_index = 0, end = spatial_dims.size(); in ShapeFromDimensions() local562 spatial_dim_index < end; ++spatial_dim_index) { in ShapeFromDimensions()564 rank, format, spatial_dim_index)] = spatial_dims[spatial_dim_index]; in ShapeFromDimensions()
407 auto check_spatial_dim = [&](int64_t spatial_dim_index, in verify()411 int64_t input_dim = input_shape[spatial_dim_index]; in verify()412 int64_t output_dim = output_shape[spatial_dim_index]; in verify()421 << spatial_dim_index << ") to be less than or equal to input " in verify()422 << dim_name << " (dimension " << spatial_dim_index in verify()429 int64_t crop_a = crops_values[2 * (spatial_dim_index - 1)]; in verify()430 int64_t crop_b = crops_values[2 * (spatial_dim_index - 1) + 1]; in verify()434 << spatial_dim_index << ") to be equal to input " << dim_name in verify()435 << " (dimension " << spatial_dim_index << ") * block_size - " in verify()