Searched refs:compute_cycles (Results 1 – 4 of 4) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorCostModel.h | 57 TensorOpCost(double bytes_loaded, double bytes_stored, double compute_cycles) in TensorOpCost() argument
60 compute_cycles_(compute_cycles) {} in TensorOpCost()
63 TensorOpCost(double bytes_loaded, double bytes_stored, double compute_cycles, in TensorOpCost() argument
67 compute_cycles_(vectorized ? compute_cycles / packet_size in TensorOpCost()
68 : compute_cycles) { in TensorOpCost()
71 eigen_assert(compute_cycles >= 0 && (numext::isfinite)(compute_cycles)); in TensorOpCost()
80 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double compute_cycles() const { in compute_cycles() function
101 double compute_cycles = numext::mini(compute_cycles_, rhs.compute_cycles()); in cwiseMin() local
102 return TensorOpCost(bytes_loaded, bytes_stored, compute_cycles); in cwiseMin()
110 double compute_cycles = numext::maxi(compute_cycles_, rhs.compute_cycles()); in cwiseMax() local
[all …]
|
| D | TensorAssign.h | 162 left.bytes_stored(), left.compute_cycles()) +
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cc | 487 int compute_cycles = Eigen::TensorOpCost::AddCost<int>() * num_boxes * 14 + in BatchedNonMaxSuppressionOp() local
493 const Eigen::TensorOpCost cost(input_bytes, output_bytes, compute_cycles); in BatchedNonMaxSuppressionOp()
526 compute_cycles = Eigen::TensorOpCost::AddCost<int>() * num_boxes * 5 + in BatchedNonMaxSuppressionOp()
531 compute_cycles); in BatchedNonMaxSuppressionOp()
566 compute_cycles = Eigen::TensorOpCost::AddCost<int>() * 2 + in BatchedNonMaxSuppressionOp()
570 compute_cycles); in BatchedNonMaxSuppressionOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops.cc | 802 const int compute_cycles = in operator ()() local
810 const Eigen::TensorOpCost cost(input_bytes, output_bytes, compute_cycles); in operator ()()
|