Searched refs:DescribeTensor (Results 1 – 2 of 2) sorted by relevance
305 static OpInfo::TensorProperties DescribeTensor(
1756 output = DescribeTensor(DT_FLOAT, {dims.batch, dims.oz, dims.oy, dims.ox}); in PredictFusedConv2DBiasActivation()1758 output = DescribeTensor(DT_FLOAT, {dims.batch, dims.oy, dims.ox, dims.oz}); in PredictFusedConv2DBiasActivation()2108 OpInfo::TensorProperties OpLevelCostEstimator::DescribeTensor( in DescribeTensor() function in tensorflow::grappler::OpLevelCostEstimator