Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc819 for (int64 spatial_dim_index = 0; in RewriteDynamicConvolutionInputGrad() local
820 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() local
888 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 …]
Ddynamic_dimension_inference.cc704 for (int64 spatial_dim_index = 0; in HandleDynamicConvolutionForward() local
705 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()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc528 for (int spatial_dim_index = 0, end = spatial_dims.size(); in DimensionsFromShape() local
529 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() local
555 spatial_dim_index < end; ++spatial_dim_index) { in ShapeFromDimensions()
557 rank, format, spatial_dim_index)] = spatial_dims[spatial_dim_index]; in ShapeFromDimensions()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc359 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()