Searched refs:conv_dims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_as_convolution_util.cc | 55 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo() local
58 {conv_dims.input_batch_dimension(), -1, in ParseConvolutionDimsInfo()
59 conv_dims.output_batch_dimension(), -1}); in ParseConvolutionDimsInfo()
61 {-1, conv_dims.kernel_output_feature_dimension(), in ParseConvolutionDimsInfo()
62 conv_dims.output_feature_dimension(), -1}); in ParseConvolutionDimsInfo()
63 dims.contracting_dims.push_back({conv_dims.input_feature_dimension(), in ParseConvolutionDimsInfo()
64 conv_dims.kernel_input_feature_dimension(), in ParseConvolutionDimsInfo()
67 for (int64 i = 0; i < conv_dims.input_spatial_dimensions_size(); ++i) { in ParseConvolutionDimsInfo()
68 int64 lhs = conv_dims.input_spatial_dimensions(i); in ParseConvolutionDimsInfo()
70 int64 rhs = conv_dims.kernel_spatial_dimensions(i); in ParseConvolutionDimsInfo()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 160 std::vector<int64> conv_dims = in Compile() local
162 conv_dims.back() = depth; in Compile()
163 conv_dims.push_back(kernel_size); in Compile()
164 conv = xla::TransposeInMinorDims(xla::Reshape(conv, conv_dims)); in Compile()
165 conv_dims.pop_back(); in Compile()
166 conv_dims.back() *= kernel_size; in Compile()
167 conv = xla::Reshape(conv, conv_dims); in Compile()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 979 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in ConvolutionDimensionsFromInputs() local
991 return conv_dims; in ConvolutionDimensionsFromInputs()
1005 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DOperations() local
1014 int64 ops = conv_dims.batch; in CountConv2DOperations()
1015 ops *= conv_dims.ox * conv_dims.oy; in CountConv2DOperations()
1016 ops *= conv_dims.kx * conv_dims.ky; in CountConv2DOperations()
1018 ops *= conv_dims.kz * conv_dims.oz; in CountConv2DOperations()
1022 conv_dims.oz *= conv_dims.iz; in CountConv2DOperations()
1023 ops *= conv_dims.oz; in CountConv2DOperations()
1028 *conv_info = conv_dims; in CountConv2DOperations()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 216 conv_dims = input_shape.ndims - num_batch_dims - 1
218 strides = [1] * conv_dims
219 elif len(strides) != conv_dims:
221 conv_dims))
222 if conv_dims == 1:
231 elif conv_dims == 2:
242 elif conv_dims == 3:
|