Created
October 31, 2018 21:09
-
-
Save Keno/4b290b1a43ae06e7a69b162a9ca9080d to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
HloModule test | |
comp1 { | |
comp1_parameter3990 = f32[] parameter(0) | |
comp1_parameter3991 = f32[] parameter(1) | |
ROOT comp1_divide3992 = f32[] divide(comp1_parameter3990, comp1_parameter3991) | |
} | |
comp2 { | |
comp2_parameter3998 = f32[] parameter(0) | |
comp2_parameter3999 = f32[] parameter(1) | |
ROOT comp2_divide4000 = f32[] divide(comp2_parameter3998, comp2_parameter3999) | |
} | |
comp3 { | |
comp3_parameter4014 = f32[] parameter(0) | |
comp3_parameter4015 = f32[] parameter(1) | |
ROOT comp3_divide4016 = f32[] divide(comp3_parameter4014, comp3_parameter4015) | |
} | |
comp4 { | |
comp4_parameter4020 = f32[] parameter(0) | |
comp4_parameter4021 = f32[] parameter(1) | |
ROOT comp4_multiply4022 = f32[] multiply(comp4_parameter4020, comp4_parameter4021) | |
} | |
comp5 { | |
comp5_parameter4024 = f32[] parameter(0) | |
comp5_parameter4025 = f32[] parameter(1) | |
ROOT comp5_subtract4026 = f32[] subtract(comp5_parameter4024, comp5_parameter4025) | |
} | |
comp6 { | |
comp6_parameter4032 = f32[] parameter(0) | |
comp6_parameter4033 = f32[] parameter(1) | |
ROOT comp6_divide4034 = f32[] divide(comp6_parameter4032, comp6_parameter4033) | |
} | |
comp8 { | |
comp8_parameter4068 = f32[] parameter(0) | |
comp8_parameter4069 = f32[] parameter(1) | |
ROOT comp8_multiply4070 = f32[] multiply(comp8_parameter4068, comp8_parameter4069) | |
} | |
comp9 { | |
comp9_parameter4072 = f32[] parameter(0) | |
comp9_parameter4073 = f32[] parameter(1) | |
ROOT comp9_subtract4074 = f32[] subtract(comp9_parameter4072, comp9_parameter4073) | |
} | |
comp10 { | |
comp10_parameter4090 = f32[] parameter(0) | |
comp10_parameter4091 = f32[] parameter(1) | |
ROOT comp10_multiply4092 = f32[] multiply(comp10_parameter4090, comp10_parameter4091) | |
} | |
comp11 { | |
comp11_parameter4094 = f32[] parameter(0) | |
comp11_parameter4095 = f32[] parameter(1) | |
ROOT comp11_subtract4096 = f32[] subtract(comp11_parameter4094, comp11_parameter4095) | |
} | |
comp12 { | |
comp12_parameter4102 = f32[] parameter(0) | |
comp12_parameter4103 = f32[] parameter(1) | |
ROOT comp12_divide4104 = f32[] divide(comp12_parameter4102, comp12_parameter4103) | |
} | |
comp7 { | |
comp7_parameter4056 = (f32[10,5]{0,1}, f32[5]{0}, f32[10]{0}, s64[], f32[10,10]{0,1}, f32[10]{0}, f32[]) parameter(0) | |
comp7_get-tuple-element4057 = f32[10,5]{0,1} get-tuple-element(comp7_parameter4056), index=0 | |
comp7_get-tuple-element4061 = f32[10,10]{0,1} get-tuple-element(comp7_parameter4056), index=4 | |
comp7_transpose4064 = f32[10,10]{0,1} transpose(comp7_get-tuple-element4061), dimensions={1,0} | |
comp7_get-tuple-element4062 = f32[10]{0} get-tuple-element(comp7_parameter4056), index=5 | |
comp7_dot4065 = f32[10]{0} dot(comp7_transpose4064, comp7_get-tuple-element4062), lhs_contracting_dims={1}, rhs_contracting_dims={0} | |
comp7_get-tuple-element4063 = f32[] get-tuple-element(comp7_parameter4056), index=6 | |
comp7_broadcast4066 = f32[10]{0} broadcast(comp7_get-tuple-element4063), dimensions={} | |
comp7_get-tuple-element4059 = f32[10]{0} get-tuple-element(comp7_parameter4056), index=2 | |
comp7_map4067 = f32[10]{0} map(comp7_broadcast4066, comp7_get-tuple-element4059), dimensions={0}, to_apply=comp8 | |
comp7_map4071 = f32[10]{0} map(comp7_dot4065, comp7_map4067), dimensions={0}, to_apply=comp9 | |
comp7_constant4076 = s64[] constant(1) | |
comp7_broadcast4077 = s64[1]{0} broadcast(comp7_constant4076), dimensions={} | |
comp7_get-tuple-element4060 = s64[] get-tuple-element(comp7_parameter4056), index=3 | |
comp7_broadcast4075 = s64[1]{0} broadcast(comp7_get-tuple-element4060), dimensions={} | |
comp7_concatenate4078 = s64[2]{0} concatenate(comp7_broadcast4077, comp7_broadcast4075), dimensions={0} | |
comp7_dynamic-slice4079 = f32[10,1]{0,1} dynamic-slice(comp7_get-tuple-element4057, comp7_concatenate4078), dynamic_slice_sizes={10,1} | |
comp7_reshape4080 = f32[10]{0} reshape(comp7_dynamic-slice4079) | |
comp7_constant4082 = s64[] constant(1) | |
comp7_broadcast4083 = s64[1]{0} broadcast(comp7_constant4082), dimensions={} | |
comp7_broadcast4081 = s64[1]{0} broadcast(comp7_get-tuple-element4060), dimensions={} | |
comp7_concatenate4084 = s64[2]{0} concatenate(comp7_broadcast4083, comp7_broadcast4081), dimensions={0} | |
comp7_dynamic-slice4085 = f32[10,1]{0,1} dynamic-slice(comp7_get-tuple-element4057, comp7_concatenate4084), dynamic_slice_sizes={10,1} | |
comp7_reshape4086 = f32[10]{0} reshape(comp7_dynamic-slice4085) | |
comp7_dot4087 = f32[] dot(comp7_reshape4086, comp7_map4071), lhs_contracting_dims={0}, rhs_contracting_dims={0} | |
comp7_broadcast4088 = f32[10]{0} broadcast(comp7_dot4087), dimensions={} | |
comp7_map4089 = f32[10]{0} map(comp7_reshape4080, comp7_broadcast4088), dimensions={0}, to_apply=comp10 | |
comp7_map4093 = f32[10]{0} map(comp7_map4071, comp7_map4089), dimensions={0}, to_apply=comp11 | |
comp7_dot4097 = f32[] dot(comp7_map4093, comp7_map4093), lhs_contracting_dims={0}, rhs_contracting_dims={0} | |
comp7_constant4098 = f32[] constant(0.5) | |
comp7_power4099 = f32[] power(comp7_dot4097, comp7_constant4098) | |
comp7_broadcast4100 = f32[10]{0} broadcast(comp7_power4099), dimensions={} | |
comp7_map4101 = f32[10]{0} map(comp7_map4093, comp7_broadcast4100), dimensions={0}, to_apply=comp12 | |
comp7_broadcast4108 = f32[10,1]{0,1} broadcast(comp7_map4101), dimensions={0} | |
comp7_constant4110 = s64[] constant(1) | |
comp7_broadcast4111 = s64[1]{0} broadcast(comp7_constant4110), dimensions={} | |
comp7_broadcast4109 = s64[1]{0} broadcast(comp7_get-tuple-element4060), dimensions={} | |
comp7_concatenate4112 = s64[2]{0} concatenate(comp7_broadcast4111, comp7_broadcast4109), dimensions={0} | |
comp7_dynamic-update-slice4113 = f32[10,5]{0,1} dynamic-update-slice(comp7_get-tuple-element4057, comp7_broadcast4108, comp7_concatenate4112) | |
comp7_get-tuple-element4058 = f32[5]{0} get-tuple-element(comp7_parameter4056), index=1 | |
comp7_broadcast4105 = f32[1]{0} broadcast(comp7_power4099), dimensions={} | |
comp7_broadcast4106 = s64[1]{0} broadcast(comp7_get-tuple-element4060), dimensions={} | |
comp7_dynamic-update-slice4107 = f32[5]{0} dynamic-update-slice(comp7_get-tuple-element4058, comp7_broadcast4105, comp7_broadcast4106) | |
comp7_constant4114 = s64[] constant(1) | |
comp7_add4115 = s64[] add(comp7_get-tuple-element4060, comp7_constant4114) | |
ROOT comp7_tuple4116 = (f32[10,5]{0,1}, f32[5]{0}, f32[10]{0}, s64[], f32[10,10]{0,1}, f32[10]{0}, f32[]) tuple(comp7_dynamic-update-slice4113, comp7_dynamic-update-slice4107, comp7_map4101, comp7_add4115, comp7_get-tuple-element4061, comp7_get-tuple-element4062, comp7_get-tuple-element4063) | |
} | |
comp13 { | |
comp13_parameter4117 = (f32[10,5]{0,1}, f32[5]{0}, f32[10]{0}, s64[], f32[10,10]{0,1}, f32[10]{0}, f32[]) parameter(0) | |
comp13_get-tuple-element4118 = f32[10,5]{0,1} get-tuple-element(comp13_parameter4117), index=0 | |
comp13_get-tuple-element4119 = f32[5]{0} get-tuple-element(comp13_parameter4117), index=1 | |
comp13_get-tuple-element4120 = f32[10]{0} get-tuple-element(comp13_parameter4117), index=2 | |
comp13_get-tuple-element4122 = f32[10,10]{0,1} get-tuple-element(comp13_parameter4117), index=4 | |
comp13_get-tuple-element4123 = f32[10]{0} get-tuple-element(comp13_parameter4117), index=5 | |
comp13_get-tuple-element4124 = f32[] get-tuple-element(comp13_parameter4117), index=6 | |
comp13_get-tuple-element4121 = s64[] get-tuple-element(comp13_parameter4117), index=3 | |
comp13_constant4125 = s64[] constant(5) | |
ROOT comp13_less-than-or-equal-to4126 = pred[] less-than-or-equal-to(comp13_get-tuple-element4121, comp13_constant4125) | |
} | |
ENTRY comp { | |
comp0_constant3976 = f32[] constant(0) | |
comp0_broadcast3977 = f32[10]{0} broadcast(comp0_constant3976), dimensions={} | |
comp0_broadcast3978 = f32[10,5]{0,1} broadcast(comp0_broadcast3977), dimensions={0} | |
comp0_parameter3971 = f32[10,10]{0,1} parameter(0) | |
comp0_transpose3986 = f32[10,10]{0,1} transpose(comp0_parameter3971), dimensions={1,0} | |
comp0_parameter3972 = f32[10]{0} parameter(1) | |
comp0_dot3987 = f32[10]{0} dot(comp0_transpose3986, comp0_parameter3972), lhs_contracting_dims={1}, rhs_contracting_dims={0} | |
comp0_dot3983 = f32[] dot(comp0_parameter3972, comp0_parameter3972), lhs_contracting_dims={0}, rhs_contracting_dims={0} | |
comp0_constant3984 = f32[] constant(0.5) | |
comp0_power3985 = f32[] power(comp0_dot3983, comp0_constant3984) | |
comp0_broadcast3988 = f32[10]{0} broadcast(comp0_power3985), dimensions={} | |
comp0_map3989 = f32[10]{0} map(comp0_dot3987, comp0_broadcast3988), dimensions={0}, to_apply=comp1 | |
comp0_dot3993 = f32[] dot(comp0_map3989, comp0_map3989), lhs_contracting_dims={0}, rhs_contracting_dims={0} | |
comp0_constant3994 = f32[] constant(0.5) | |
comp0_power3995 = f32[] power(comp0_dot3993, comp0_constant3994) | |
comp0_broadcast3996 = f32[10]{0} broadcast(comp0_power3995), dimensions={} | |
comp0_map3997 = f32[10]{0} map(comp0_map3989, comp0_broadcast3996), dimensions={0}, to_apply=comp2 | |
comp0_broadcast4002 = f32[10,1]{0,1} broadcast(comp0_map3997), dimensions={0} | |
comp0_constant4004 = s64[] constant(1) | |
comp0_broadcast4005 = s64[1]{0} broadcast(comp0_constant4004), dimensions={} | |
comp0_constant4001 = s64[] constant(1) | |
comp0_broadcast4003 = s64[1]{0} broadcast(comp0_constant4001), dimensions={} | |
comp0_concatenate4006 = s64[2]{0} concatenate(comp0_broadcast4005, comp0_broadcast4003), dimensions={0} | |
comp0_dynamic-update-slice4007 = f32[10,5]{0,1} dynamic-update-slice(comp0_broadcast3978, comp0_broadcast4002, comp0_concatenate4006) | |
comp0_constant3979 = f32[] constant(0) | |
comp0_broadcast3980 = f32[5]{0} broadcast(comp0_constant3979), dimensions={} | |
comp0_broadcast4009 = f32[1]{0} broadcast(comp0_power3995), dimensions={} | |
comp0_constant4008 = s64[] constant(1) | |
comp0_broadcast4010 = s64[1]{0} broadcast(comp0_constant4008), dimensions={} | |
comp0_dynamic-update-slice4011 = f32[5]{0} dynamic-update-slice(comp0_broadcast3980, comp0_broadcast4009, comp0_broadcast4010) | |
comp0_constant4053 = s64[] constant(2) | |
comp0_dot4017 = f32[10]{0} dot(comp0_parameter3971, comp0_map3997), lhs_contracting_dims={1}, rhs_contracting_dims={0} | |
comp0_broadcast4018 = f32[10]{0} broadcast(comp0_power3995), dimensions={} | |
comp0_broadcast4012 = f32[10]{0} broadcast(comp0_power3985), dimensions={} | |
comp0_map4013 = f32[10]{0} map(comp0_parameter3972, comp0_broadcast4012), dimensions={0}, to_apply=comp3 | |
comp0_map4019 = f32[10]{0} map(comp0_broadcast4018, comp0_map4013), dimensions={0}, to_apply=comp4 | |
comp0_map4023 = f32[10]{0} map(comp0_dot4017, comp0_map4019), dimensions={0}, to_apply=comp5 | |
comp0_dot4027 = f32[] dot(comp0_map4023, comp0_map4023), lhs_contracting_dims={0}, rhs_contracting_dims={0} | |
comp0_constant4028 = f32[] constant(0.5) | |
comp0_power4029 = f32[] power(comp0_dot4027, comp0_constant4028) | |
comp0_broadcast4030 = f32[10]{0} broadcast(comp0_power4029), dimensions={} | |
comp0_map4031 = f32[10]{0} map(comp0_map4023, comp0_broadcast4030), dimensions={0}, to_apply=comp6 | |
comp0_tuple4054 = (f32[10,5]{0,1}, f32[5]{0}, f32[10]{0}, s64[], f32[10,10]{0,1}, f32[10]{0}, f32[]) tuple(comp0_dynamic-update-slice4007, comp0_dynamic-update-slice4011, comp0_map3997, comp0_constant4053, comp0_parameter3971, comp0_map4031, comp0_power4029) | |
comp0_while4055 = (f32[10,5]{0,1}, f32[5]{0}, f32[10]{0}, s64[], f32[10,10]{0,1}, f32[10]{0}, f32[]) while(comp0_tuple4054), condition=comp13, body=comp7 | |
comp0_get-tuple-element4127 = f32[5]{0} get-tuple-element(comp0_while4055), index=1 | |
comp0_constant3981 = f32[] constant(0) | |
comp0_broadcast3982 = f32[5]{0} broadcast(comp0_constant3981), dimensions={} | |
comp0_broadcast4050 = f32[1]{0} broadcast(comp0_power4029), dimensions={} | |
comp0_constant4049 = s64[] constant(1) | |
comp0_broadcast4051 = s64[1]{0} broadcast(comp0_constant4049), dimensions={} | |
comp0_dynamic-update-slice4052 = f32[5]{0} dynamic-update-slice(comp0_broadcast3982, comp0_broadcast4050, comp0_broadcast4051) | |
comp0_constant3973 = f32[] constant(0) | |
comp0_broadcast3974 = f32[10]{0} broadcast(comp0_constant3973), dimensions={} | |
comp0_broadcast3975 = f32[10,6]{0,1} broadcast(comp0_broadcast3974), dimensions={0} | |
comp0_broadcast4036 = f32[10,1]{0,1} broadcast(comp0_map4013), dimensions={0} | |
comp0_constant4038 = s64[] constant(1) | |
comp0_broadcast4039 = s64[1]{0} broadcast(comp0_constant4038), dimensions={} | |
comp0_constant4035 = s64[] constant(1) | |
comp0_broadcast4037 = s64[1]{0} broadcast(comp0_constant4035), dimensions={} | |
comp0_concatenate4040 = s64[2]{0} concatenate(comp0_broadcast4039, comp0_broadcast4037), dimensions={0} | |
comp0_dynamic-update-slice4041 = f32[10,6]{0,1} dynamic-update-slice(comp0_broadcast3975, comp0_broadcast4036, comp0_concatenate4040) | |
comp0_broadcast4043 = f32[10,1]{0,1} broadcast(comp0_map4031), dimensions={0} | |
comp0_constant4045 = s64[] constant(1) | |
comp0_broadcast4046 = s64[1]{0} broadcast(comp0_constant4045), dimensions={} | |
comp0_constant4042 = s64[] constant(2) | |
comp0_broadcast4044 = s64[1]{0} broadcast(comp0_constant4042), dimensions={} | |
comp0_concatenate4047 = s64[2]{0} concatenate(comp0_broadcast4046, comp0_broadcast4044), dimensions={0} | |
comp0_dynamic-update-slice4048 = f32[10,6]{0,1} dynamic-update-slice(comp0_dynamic-update-slice4041, comp0_broadcast4043, comp0_concatenate4047) | |
comp0_get-tuple-element4128 = f32[10,5]{0,1} get-tuple-element(comp0_while4055), index=0 | |
ROOT comp0_tuple4129 = (f32[5]{0}, f32[5]{0}, f32[10,6]{0,1}, f32[10,5]{0,1}) tuple(comp0_get-tuple-element4127, comp0_dynamic-update-slice4052, comp0_dynamic-update-slice4048, comp0_get-tuple-element4128) | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment