Searched refs:OpDimensionsFromInputs (Results 1 – 3 of 3) sorted by relevance
293 static ConvolutionDimensions OpDimensionsFromInputs(
2123 OpLevelCostEstimator::OpDimensionsFromInputs( in OpDimensionsFromInputs() function in tensorflow::grappler::OpLevelCostEstimator2175 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictMaxPool()2216 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictMaxPoolGrad()2292 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictAvgPool()2350 OpDimensionsFromInputs(x_shape, op_info, &found_unknown_shapes); in PredictAvgPoolGrad()2376 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictFusedBatchNorm()2426 ConvolutionDimensions dims = OpDimensionsFromInputs( in PredictFusedBatchNormGrad()
561 auto dims = OpLevelCostEstimator::OpDimensionsFromInputs( in ValidateOpDimensionsFromInputs()1372 TEST_F(OpLevelCostEstimatorTest, OpDimensionsFromInputs) { in TEST_F() argument