Searched refs:CalculateTensorElementCount (Results 1 – 2 of 2) sorted by relevance
785 CalculateTensorElementCount(op_info.outputs(0), &found_unknown_shapes)); in PredictCwiseOp()1527 int64 OpLevelCostEstimator::CalculateTensorElementCount( in CalculateTensorElementCount() function in tensorflow::grappler::OpLevelCostEstimator1543 int64 count = CalculateTensorElementCount(tensor, found_unknown_shapes); in CalculateTensorSize()1577 CalculateTensorElementCount(input, found_unknown_shapes); in CalculateLargestInputCount()1836 CalculateTensorElementCount(op_info.inputs(1), &found_unknown_shapes); in PredictSparseTensorDenseMatMul()1968 CalculateTensorElementCount(op_info.outputs(0), &unknown_shapes); in PredictGatherOrSlice()1994 CalculateTensorElementCount(op_info.inputs(i), &unknown_shapes)); in PredictGatherOrSlice()2017 CalculateTensorElementCount(op_info.inputs(1), &found_unknown_shapes); in PredictScatter()2271 : CalculateTensorElementCount(op_info.inputs(1), in PredictAssignVariableOps()2465 CalculateTensorElementCount(op_info.outputs(0), &found_unknown_shapes)); in PredictNaryOp()[all …]
256 static int64 CalculateTensorElementCount(