Searched refs:out_cast (Results 1 – 6 of 6) sorted by relevance
146 Array out_cast = (x_cast - mean_cast) * inv_std * gamma_cast + beta_cast; in ApplyGenericBatchNorm() local147 const Array& actual_out = out_cast.dtype() == x.dtype() ? out_cast : out_cast.AsType(x.dtype()); in ApplyGenericBatchNorm()
45 ; CHECK: %[[out_cast:[^=]+]] = bitcast [8 x i64]* %out to i8*46 ; CHECK: call void @llvm.memcpy.p0i8.p0i8.i64(i8* nonnull align 8 %[[out_cast]], i8* nonnul…
45 ; CHECK: %[[out_cast:[^=]+]] = bitcast [8 x i64]* %out to i8*46 ; CHECK: call void @llvm.memcpy.p0i8.p0i8.i64(i8* align 8 %[[out_cast]], i8* align 8 %[[a_c…
722 const auto &out_cast = dynamic_cast<const logic_output_t &>(out); in get_d_a_proxy() local731 out_cast.logic_family()->create_d_a_proxy(m_nlstate, x, &out_cast); in get_d_a_proxy()