Searched refs:spatial_dim_index (Results 1 – 4 of 4) sorted by relevance
819 for (int64 spatial_dim_index = 0; in RewriteDynamicConvolutionInputGrad() local820 spatial_dim_index < dnums.input_spatial_dimensions_size(); in RewriteDynamicConvolutionInputGrad()821 ++spatial_dim_index) { in RewriteDynamicConvolutionInputGrad()822 int64 input_spatial_dim = dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad()835 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad()842 padding_before[spatial_dim_index] = dynamic_window_dims.padding_before; in RewriteDynamicConvolutionInputGrad()887 for (int64 spatial_dim_index = 0; in RewriteDynamicConvolutionForward() local888 spatial_dim_index < dnums.input_spatial_dimensions_size(); in RewriteDynamicConvolutionForward()889 ++spatial_dim_index) { in RewriteDynamicConvolutionForward()890 int64 input_spatial_dim = dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionForward()[all …]
704 for (int64 spatial_dim_index = 0; in HandleDynamicConvolutionForward() local705 spatial_dim_index < dimension_numbers.input_spatial_dimensions_size(); in HandleDynamicConvolutionForward()706 ++spatial_dim_index) { in HandleDynamicConvolutionForward()708 dimension_numbers.input_spatial_dimensions(spatial_dim_index); in HandleDynamicConvolutionForward()710 dimension_numbers.output_spatial_dimensions(spatial_dim_index); in HandleDynamicConvolutionForward()713 WindowDimension window_dim = hlo->window().dimensions(spatial_dim_index); in HandleDynamicConvolutionForward()
528 for (int spatial_dim_index = 0, end = spatial_dims.size(); in DimensionsFromShape() local529 spatial_dim_index < end; ++spatial_dim_index) { in DimensionsFromShape()530 spatial_dims[spatial_dim_index] = context->Dim( in DimensionsFromShape()531 shape, GetTensorSpatialDimIndex(rank, format, spatial_dim_index)); in DimensionsFromShape()554 for (int spatial_dim_index = 0, end = spatial_dims.size(); in ShapeFromDimensions() local555 spatial_dim_index < end; ++spatial_dim_index) { in ShapeFromDimensions()557 rank, format, spatial_dim_index)] = spatial_dims[spatial_dim_index]; in ShapeFromDimensions()
359 auto check_spatial_dim = [&](int64_t spatial_dim_index, in Verify()363 int64_t input_dim = input_shape[spatial_dim_index]; in Verify()364 int64_t output_dim = output_shape[spatial_dim_index]; in Verify()373 << spatial_dim_index << ") to be less than or equal to input " in Verify()374 << dim_name << " (dimension " << spatial_dim_index in Verify()381 int64_t crop_a = crops_values[2 * (spatial_dim_index - 1)]; in Verify()382 int64_t crop_b = crops_values[2 * (spatial_dim_index - 1) + 1]; in Verify()386 << spatial_dim_index << ") to be equal to input " << dim_name in Verify()387 << " (dimension " << spatial_dim_index << ") * block_size - " in Verify()