Searched defs:magnitude_type (Results 1 – 4 of 4) sorted by relevance
46 void setup(TensorShape shape, Format format, MagnitudeType magnitude_type) in setup()60 TensorType compute_target(const TensorShape &shape, Format format, MagnitudeType magnitude_type) in compute_target()101 …eTensor<T> compute_reference(const TensorShape &shape, Format format, MagnitudeType magnitude_type) in compute_reference()
43 AbsoluteTolerance<T> tolerance(MagnitudeType magnitude_type) in tolerance()
35 …or<T> magnitude(const SimpleTensor<T> &gx, const SimpleTensor<T> &gy, MagnitudeType magnitude_type) in magnitude()