Home
last modified time | relevance | path

Searched defs:conv_dims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dextract_image_patches_op.cc160 std::vector<int64> conv_dims = in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_as_convolution_util.cc81 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc990 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in ConvolutionDimensionsFromInputs() local
1016 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DOperations() local
1459 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropInputOperations() local
1516 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropFilterOperations() local
2178 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in OpDimensionsFromInputs() local