Searched refs:while_value (Results 1 – 2 of 2) 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()257 const HloValue& while_value = dataflow_.GetUniqueValueAt( in ComputeWhileAliasedBuffers() local263 << while_value.ToShortString(); in ComputeWhileAliasedBuffers()264 aliased_buffers->push_back(GetBufferForValue(while_value)); in ComputeWhileAliasedBuffers()
428 ans, pfor_value, while_value = sess.run(432 self.assertAllClose(ans, while_value)477 ans, pfor_value, while_value = sess.run(481 self.assertAllClose(ans, while_value)