Searched refs:CalculateTensorElementCount (Results 1 – 2 of 2) sorted by relevance
460 CalculateTensorElementCount(op_info.outputs(0), &found_unknown_shapes)); in PredictCwiseOp()1030 int64 OpLevelCostEstimator::CalculateTensorElementCount( in CalculateTensorElementCount() function in tensorflow::grappler::OpLevelCostEstimator1046 int64 count = CalculateTensorElementCount(tensor, found_unknown_shapes); in CalculateTensorSize()1069 CalculateTensorElementCount(input, found_unknown_shapes); in CalculateLargestInputCount()1245 CalculateTensorElementCount(op_info.inputs(1), &found_unknown_shapes); in PredictSparseTensorDenseMatMul()1350 CalculateTensorElementCount(op_info.outputs(0), &unknown_shapes); in PredictGatherOrSlice()1357 CalculateTensorElementCount(op_info.inputs(1), &unknown_shapes) + in PredictGatherOrSlice()1358 CalculateTensorElementCount(op_info.inputs(2), &unknown_shapes); in PredictGatherOrSlice()1362 CalculateTensorElementCount(op_info.inputs(1), &unknown_shapes); in PredictGatherOrSlice()
97 int64 CalculateTensorElementCount(const OpInfo::TensorProperties& tensor,