Searched refs:trailing_dims_product (Results 1 – 1 of 1) sorted by relevance
146 int64 trailing_dims_product = 1; in GetThreeFryInputsAndUpdatedState() local153 ConstantR0<uint64>(builder, trailing_dims_product)); in GetThreeFryInputsAndUpdatedState()154 trailing_dims_product *= shape.dimensions(i); in GetThreeFryInputsAndUpdatedState()