Home
last modified time | relevance | path

Searched refs:ConvolutionDimensions (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.h64 struct ConvolutionDimensions { struct
81 ConvolutionDimensions* conv_info, argument
90 const OpInfo& op_info, ConvolutionDimensions* returned_conv_dims,
93 const OpInfo& op_info, ConvolutionDimensions* returned_conv_dims,
164 static ConvolutionDimensions ConvolutionDimensionsFromInputs(
170 static ConvolutionDimensions OpDimensionsFromInputs(
Dop_level_cost_estimator.cc571 OpLevelCostEstimator::ConvolutionDimensions
632 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in ConvolutionDimensionsFromInputs()
647 const OpInfo& op_info, ConvolutionDimensions* conv_info, in CountConv2DOperations()
652 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DOperations()
917 const OpInfo& op_info, ConvolutionDimensions* returned_conv_dims, in CountConv2DBackpropInputOperations()
951 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropInputOperations()
975 const OpInfo& op_info, ConvolutionDimensions* returned_conv_dims, in CountConv2DBackpropFilterOperations()
1008 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropFilterOperations()
1438 OpLevelCostEstimator::ConvolutionDimensions
1479 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in OpDimensionsFromInputs()
[all …]