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