Searched refs:CalculateTensorElementCount (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator.cc | 797 CalculateTensorElementCount(op_info.outputs(0), &found_unknown_shapes)); in PredictCwiseOp()
1539 int64_t OpLevelCostEstimator::CalculateTensorElementCount( in CalculateTensorElementCount() function in tensorflow::grappler::OpLevelCostEstimator
1562 int64_t count = CalculateTensorElementCount(tensor, found_unknown_shapes); in CalculateTensorSize()
1602 CalculateTensorElementCount(input, found_unknown_shapes); in CalculateLargestInputCount()
1870 CalculateTensorElementCount(op_info.inputs(1), &found_unknown_shapes); in PredictSparseTensorDenseMatMul()
2002 CalculateTensorElementCount(op_info.outputs(0), &unknown_shapes); in PredictGatherOrSlice()
2028 CalculateTensorElementCount(op_info.inputs(i), &unknown_shapes)); in PredictGatherOrSlice()
2051 CalculateTensorElementCount(op_info.inputs(1), &found_unknown_shapes); in PredictScatter()
2313 : CalculateTensorElementCount(op_info.inputs(1), in PredictAssignVariableOps()
2512 CalculateTensorElementCount(op_info.outputs(0), &found_unknown_shapes)); in PredictNaryOp()
[all …]
|
| H A D | op_level_cost_estimator.h | 256 static int64_t CalculateTensorElementCount(
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2022-038.md | 12 int64_t count = CalculateTensorElementCount(tensor, found_unknown_shapes);
|