Searched refs:GetBufferForValue (Results 1 – 1 of 1) sorted by relevance
171 BufferNumber GetBufferForValue(const HloValue& value) { in GetBufferForValue() function in xla::BufferValueMap198 GetBufferForValue(get_parameter_value(*aliased_input))); in ComputeInputOutputAliasedBuffers()206 aliased_buffers->push_back(GetBufferForValue(value)); in ComputeInputOutputAliasedBuffers()219 aliased_buffers->push_back(GetBufferForValue(while_value)); in ComputeWhileAliasedBuffers()241 aliased_buffers->push_back(GetBufferForValue(while_value)); in ComputeWhileAliasedBuffers()265 aliased_buffers->push_back(GetBufferForValue(while_value)); in ComputeWhileAliasedBuffers()273 aliased_buffers->push_back(GetBufferForValue(value)); in ComputeWhileAliasedBuffers()299 aliased_buffers->push_back(GetBufferForValue(cond_value)); in ComputeConditionalAliasedBuffers()307 aliased_buffers->push_back(GetBufferForValue(value)); in ComputeConditionalAliasedBuffers()326 aliased_buffers->push_back(GetBufferForValue(operand_value)); in ComputeInPlaceOperationAliasedBuffers()[all …]