Searched refs:OpDimensionsFromInputs (Results 1 – 3 of 3) sorted by relevance
1439 OpLevelCostEstimator::OpDimensionsFromInputs( in OpDimensionsFromInputs() function in tensorflow::grappler::OpLevelCostEstimator1488 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictMaxPool()1525 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictMaxPoolGrad()1565 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictAvgPool()1622 OpDimensionsFromInputs(x_shape, op_info, &found_unknown_shapes); in PredictAvgPoolGrad()1656 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictFusedBatchNorm()1703 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictFusedBatchNormGrad()
170 static ConvolutionDimensions OpDimensionsFromInputs(
492 auto dims = OpLevelCostEstimator::OpDimensionsFromInputs( in ValidateOpDimensionsFromImputs()995 TEST_F(OpLevelCostEstimatorTest, OpDimensionsFromInputs) { in TEST_F() argument