Searched refs:while_value (Results 1 – 3 of 3) sorted by relevance
217 const HloValue& while_value = in ComputeWhileAliasedBuffers() local219 aliased_buffers->push_back(GetBufferForValue(while_value)); in ComputeWhileAliasedBuffers()222 << while_value.ToShortString(); in ComputeWhileAliasedBuffers()236 const HloValue& while_value = dataflow_.GetUniqueValueAt( in ComputeWhileAliasedBuffers() local240 << while_value.ToShortString(); in ComputeWhileAliasedBuffers()241 aliased_buffers->push_back(GetBufferForValue(while_value)); in ComputeWhileAliasedBuffers()258 const HloValue& while_value = dataflow_.GetUniqueValueAt( in ComputeWhileAliasedBuffers() local264 << while_value.ToShortString(); in ComputeWhileAliasedBuffers()265 aliased_buffers->push_back(GetBufferForValue(while_value)); in ComputeWhileAliasedBuffers()
932 HloValue* while_value = in IsUseAllowedInAlternateMemory() local937 RequiredMemoryAssignmentAt(while_value, while_time); in IsUseAllowedInAlternateMemory()
427 ans, pfor_value, while_value = sess.run(431 self.assertAllClose(ans, while_value)476 ans, pfor_value, while_value = sess.run(480 self.assertAllClose(ans, while_value)