Home
last modified time | relevance | path

Searched refs:spatial_dim (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_as_convolution_util.cc116 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvForDotGeneralConvolution()
118 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
123 if (dim.spatial_dim < 0) { in CreateShardedConvForDotGeneralConvolution()
126 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvForDotGeneralConvolution()
128 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
131 if (dim.spatial_dim < 0) { in CreateShardedConvForDotGeneralConvolution()
134 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvForDotGeneralConvolution()
136 conv_dnums.kernel_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
163 dnums.batch_dims.back().spatial_dim = -1; in ParseDotGeneralFromDot()
173 dnums.contracting_dims.back().spatial_dim = -1; in ParseDotGeneralFromDot()
[all …]
Dconvolution_4d_expander.cc46 int64 spatial_dim = dim_nums.input_spatial_dimensions(i); in InstructionMatchesPattern() local
47 if (input.dimensions(spatial_dim) == 1 && in InstructionMatchesPattern()
Ddot_as_convolution_util.h41 int64 spatial_dim; member
Dalgebraic_simplifier.cc5294 for (int64 spatial_dim = 0; in SwapConvOperands() local
5295 spatial_dim < dnums.input_spatial_dimensions_size(); ++spatial_dim) { in SwapConvOperands()
5296 const int64 kernel_size = window_dims[spatial_dim].size(); in SwapConvOperands()
5298 !window_dims[spatial_dim].window_reversal() && in SwapConvOperands()
5299 window_dims[spatial_dim].padding_low() == 0 && in SwapConvOperands()
5300 window_dims[spatial_dim].padding_high() == 0 && in SwapConvOperands()
5301 window_dims[spatial_dim].window_dilation() == 1; in SwapConvOperands()
5304 window_dims[spatial_dim].base_dilation() == kernel_size && in SwapConvOperands()
5305 window_dims[spatial_dim].stride() == kernel_size - 1; in SwapConvOperands()
5307 input->shape().dimensions(dnums.input_spatial_dimensions(spatial_dim)); in SwapConvOperands()
[all …]
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h227 int spatial_dim) { in GetTensorSpatialDimIndex() argument
228 CHECK(spatial_dim >= 0 && in GetTensorSpatialDimIndex()
229 spatial_dim < GetTensorSpatialDims(num_dims, format)) in GetTensorSpatialDimIndex()
230 << spatial_dim << " " << num_dims << " " << ToString(format); in GetTensorSpatialDimIndex()
234 return spatial_dim + 1; in GetTensorSpatialDimIndex()
237 return spatial_dim + 2; in GetTensorSpatialDimIndex()
240 return spatial_dim; in GetTensorSpatialDimIndex()
603 for (int spatial_dim = 0; spatial_dim < num_src_spatial_dims; ++spatial_dim) { in ShapeFromFormat() local
604 spatial_dims[spatial_dim] = in ShapeFromFormat()
606 src_shape.dims(), src_format, spatial_dim)]; in ShapeFromFormat()
Dtensor_format_test.cc50 int spatial(int spatial_index) const { return spatial_dim[spatial_index]; } in spatial()
53 int spatial_dim[3]; member
62 int spatial(int spatial_index) const { return spatial_dim[spatial_index]; } in spatial()
65 int spatial_dim[3]; member
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc57 Padding padding, int64 padding_before, int64 padding_after, int spatial_dim, in ConvBackpropExtractAndVerifyDimension() argument
59 dim->input_size = input_shape.dim_size(spatial_dim); in ConvBackpropExtractAndVerifyDimension()
61 dim->output_size = output_shape.dim_size(spatial_dim); in ConvBackpropExtractAndVerifyDimension()
62 dim->stride = strides[spatial_dim]; in ConvBackpropExtractAndVerifyDimension()
63 dim->dilation = dilations[spatial_dim]; in ConvBackpropExtractAndVerifyDimension()
72 " spatial_dim: ", spatial_dim, " input: ", dim->input_size, in ConvBackpropExtractAndVerifyDimension()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc902 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvConvolution()
904 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvConvolution()
909 if (dim.spatial_dim < 0) { in CreateShardedConvConvolution()
912 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvConvolution()
914 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvConvolution()
917 if (dim.spatial_dim < 0) { in CreateShardedConvConvolution()
920 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvConvolution()
922 conv_dnums.kernel_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvConvolution()
928 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvConvolution()
929 const auto& new_window_dimension = conv_window.dimensions(dim.spatial_dim); in CreateShardedConvConvolution()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc143 for (int spatial_dim = 0; spatial_dim < num_spatial_dims; ++spatial_dim) { in XlaTensorFormat() local
144 spatial_dimensions[spatial_dim] = in XlaTensorFormat()
145 GetTensorSpatialDimIndex(num_dims, data_format, spatial_dim); in XlaTensorFormat()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc510 for (int spatial_dim = 0, end = spatial.size(); spatial_dim < end; in MakeShapeFromFormat() local
511 spatial_dim++) { in MakeShapeFromFormat()
512 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] = in MakeShapeFromFormat()
513 context->MakeDim(spatial[spatial_dim]); in MakeShapeFromFormat()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h254 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } in spatial_dim() function