- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetArithmeticCountForConvAndFullyconnectedOp (0.47 sec)
-
tensorflow/compiler/mlir/lite/utils/arithmetic_count_util.h
} *count = total_count; return true; } // For conv2d/depthwise_conv/fully_connected ops. // This algorithm actually comes from TOCO tooling_util.cc static bool GetArithmeticCountForConvAndFullyconnectedOp(mlir::Operation* op, int64_t* count) { auto weight = op->getOperand(1); auto weight_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
results.add<RemoveOptionalZeroBias<FullyConnectedOp>>(context); } int64_t FullyConnectedOp::GetArithmeticCount(Operation* op) { int64_t count; if (ArithmeticCountUtilHelper::GetArithmeticCountForConvAndFullyconnectedOp( op, &count)) return count; return -1; } //===----------------------------------------------------------------------===// // Conv2DOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)