Searched refs:spatial_dim (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | dot_as_convolution_util.cc | 116 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 …]
|
D | convolution_4d_expander.cc | 46 int64 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 | 41 int64 spatial_dim; member
|
D | algebraic_simplifier.cc | 5294 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/ |
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 | 57 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/ |
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 | 143 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/ |
D | common_shape_fns.cc | 510 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/ |
D | dnn.h | 254 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } in spatial_dim() function
|