Searched refs:GetFirstOutputCount (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 711 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) return count; in GetArithmeticCount()
1533 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) return count; in GetArithmeticCount()
1552 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) return count; in GetArithmeticCount()
2166 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) return count; in GetArithmeticCount()
3688 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) in GetArithmeticCount()
3704 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) in GetArithmeticCount()
3720 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) in GetArithmeticCount()
3736 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) in GetArithmeticCount()
3748 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) { in GetArithmeticCount()
3763 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) { in GetArithmeticCount()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | arithmetic_count_util.h | 27 static bool GetFirstOutputCount(mlir::Operation* op, int64_t* count) { in GetFirstOutputCount() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/hardwares/ |
| H A D | cpu_hardware.cc | 126 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) { in GetOpCost()
|
| H A D | gpu_hardware.cc | 102 if (ArithmeticCountUtilHelper::GetFirstOutputCount(op, &count)) in GetOpCost()
|