Home
last modified time | relevance | path

Searched defs:compute_cycles (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorCostModel.h57 TensorOpCost(double bytes_loaded, double bytes_stored, double compute_cycles) in TensorOpCost()
63 TensorOpCost(double bytes_loaded, double bytes_stored, double 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
110 double compute_cycles = numext::maxi(compute_cycles_, rhs.compute_cycles()); in cwiseMax() local
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_overlap.py35 def __init__(self, compute_cycles, has_params: bool): argument
61 def _create_model(compute_cycles, has_params: bool): argument
104 def run(compute_cycles, all_gather_cycles): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dresize_nearest_neighbor_op.cc201 const int compute_cycles = (Eigen::TensorOpCost::ModCost<T>() * 2 + in operator ()() local
H A Dnon_max_suppression_op.cc510 int compute_cycles = Eigen::TensorOpCost::AddCost<int>() * num_boxes * 14 + in BatchedNonMaxSuppressionOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops.cc830 const int compute_cycles = in operator ()() local