Home
last modified time | relevance | path

Searched refs:spatial_dim_index (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_padder.cc973 for (int64_t spatial_dim_index = 0; in RewriteDynamicConvolutionInputGrad() local
974 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() local
1044 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 …]
H A Ddynamic_dimension_inference.cc752 for (int64_t spatial_dim_index = 0; in HandleDynamicConvolutionForward() local
753 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc532 for (int spatial_dim_index = 0, end = spatial_dims.size(); in DimensionsFromShape() local
533 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() local
562 spatial_dim_index < end; ++spatial_dim_index) { in ShapeFromDimensions()
564 rank, format, spatial_dim_index)] = spatial_dims[spatial_dim_index]; in ShapeFromDimensions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc407 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()