| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_as_convolution_util.cc | 142 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvForDotGeneralConvolution()
144 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
149 if (dim.spatial_dim < 0) { in CreateShardedConvForDotGeneralConvolution()
152 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvForDotGeneralConvolution()
154 conv_dnums.input_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
157 if (dim.spatial_dim < 0) { in CreateShardedConvForDotGeneralConvolution()
160 auto wd = window.mutable_dimensions(dim.spatial_dim); in CreateShardedConvForDotGeneralConvolution()
162 conv_dnums.kernel_spatial_dimensions(dim.spatial_dim))); in CreateShardedConvForDotGeneralConvolution()
189 dnums.batch_dims.back().spatial_dim = -1; in ParseDotGeneralFromDot()
199 dnums.contracting_dims.back().spatial_dim = -1; in ParseDotGeneralFromDot()
[all …]
|
| D | convolution_4d_expander.cc | 46 int64_t spatial_dim = dim_nums.input_spatial_dimensions(i); in InstructionMatchesPattern() local
47 if (input.dimensions(spatial_dim) == 1 && in InstructionMatchesPattern()
|
| D | dot_as_convolution_util.h | 51 int64 spatial_dim; member
|
| D | algebraic_simplifier.cc | 5607 for (int64_t spatial_dim = 0; in SwapConvOperands() local
5608 spatial_dim < dnums.input_spatial_dimensions_size(); ++spatial_dim) { in SwapConvOperands()
5609 const int64_t kernel_size = window_dims[spatial_dim].size(); in SwapConvOperands()
5611 !window_dims[spatial_dim].window_reversal() && in SwapConvOperands()
5612 window_dims[spatial_dim].padding_low() == 0 && in SwapConvOperands()
5613 window_dims[spatial_dim].padding_high() == 0 && in SwapConvOperands()
5614 window_dims[spatial_dim].window_dilation() == 1; in SwapConvOperands()
5617 window_dims[spatial_dim].base_dilation() == kernel_size && in SwapConvOperands()
5618 window_dims[spatial_dim].stride() == kernel_size - 1; in SwapConvOperands()
5620 input->shape().dimensions(dnums.input_spatial_dimensions(spatial_dim)); in SwapConvOperands()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 227 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()
|
| D | tensor_format_test.cc | 50 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/ |
| D | conv_grad_shape_utils.cc | 58 int spatial_dim, int filter_spatial_dim, in ConvBackpropExtractAndVerifyDimension() argument
60 dim->input_size = input_shape.dim_size(spatial_dim); in ConvBackpropExtractAndVerifyDimension()
62 dim->output_size = output_shape.dim_size(spatial_dim); in ConvBackpropExtractAndVerifyDimension()
63 dim->stride = strides[spatial_dim]; in ConvBackpropExtractAndVerifyDimension()
64 dim->dilation = dilations[spatial_dim]; in ConvBackpropExtractAndVerifyDimension()
73 " spatial_dim: ", spatial_dim, " input: ", dim->input_size, in ConvBackpropExtractAndVerifyDimension()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops_base_structs.cc | 64 for (auto spatial_dim : in printConvolutionDimensions() local
66 dims[spatial_dim.value()] = static_cast<int64_t>(spatial_dim.index()); in printConvolutionDimensions()
120 int64_t spatial_dim; in parseConvolutionDimensions() local
122 parser.parseOptionalInteger(spatial_dim); in parseConvolutionDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 902 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/ |
| D | pooling_ops.cc | 149 for (int spatial_dim = 0; spatial_dim < num_spatial_dims; ++spatial_dim) { in XlaTensorFormat() local
150 spatial_dimensions[spatial_dim] = in XlaTensorFormat()
151 GetTensorSpatialDimIndex(num_dims, data_format, spatial_dim); in XlaTensorFormat()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 513 for (int spatial_dim = 0, end = spatial.size(); spatial_dim < end; in MakeShapeFromFormat() local
514 spatial_dim++) { in MakeShapeFromFormat()
515 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] = in MakeShapeFromFormat()
516 context->MakeDim(spatial[spatial_dim]); in MakeShapeFromFormat()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 268 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } in spatial_dim() function
|