Searched defs:compute_cycles (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
H A D | TensorCostModel.h | 57 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 D | test_fsdp_overlap.py | 35 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 D | resize_nearest_neighbor_op.cc | 201 const int compute_cycles = (Eigen::TensorOpCost::ModCost<T>() * 2 + in operator ()() local
|
H A D | non_max_suppression_op.cc | 510 int compute_cycles = Eigen::TensorOpCost::AddCost<int>() * num_boxes * 14 + in BatchedNonMaxSuppressionOp() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | training_ops.cc | 830 const int compute_cycles = in operator ()() local
|