Searched refs:bytes_stored (Results 1 – 3 of 3) 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 59 bytes_stored_(bytes_stored), in TensorOpCost() 63 TensorOpCost(double bytes_loaded, double bytes_stored, double compute_cycles, in TensorOpCost() argument 66 bytes_stored_(bytes_stored), in TensorOpCost() 70 eigen_assert(bytes_stored >= 0 && (numext::isfinite)(bytes_stored)); in TensorOpCost() 77 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double bytes_stored() const { in bytes_stored() function 100 double bytes_stored = numext::mini(bytes_stored_, rhs.bytes_stored()); in cwiseMin() local 102 return TensorOpCost(bytes_loaded, bytes_stored, compute_cycles); in cwiseMin() 109 double bytes_stored = numext::maxi(bytes_stored_, rhs.bytes_stored()); in cwiseMax() local 111 return TensorOpCost(bytes_loaded, bytes_stored, compute_cycles); in cwiseMax() [all …]
|
D | TensorAssign.h | 162 left.bytes_stored(), left.compute_cycles()) +
|
/external/tensorflow/tensorflow/core/kernels/ |
D | one_hot_op.h | 91 double bytes_stored = sizeof(T); 93 const Eigen::TensorOpCost cost(bytes_loaded, bytes_stored, cycles);
|