Searched refs:is_entry_parameter (Results 1 – 6 of 6) sorted by relevance
48 bool is_entry_parameter() const { return kind() == Kind::kEntryParameter; } in is_entry_parameter() function52 assert(is_entry_parameter()); in entry_parameter_number()82 return !is_entry_parameter() ||
64 (buffer_infos[i].is_entry_parameter() && allocate_entry_params); in AlignedBufferBytes()91 (buffer_infos[i].is_entry_parameter() && allocate_entry_params); in MallocContiguousBuffers()
47 if (buffer_infos[i].is_entry_parameter()) { in CreateArgIndexTableFromBufferInfos()
103 CHECK(result->buffer_infos()[0].is_entry_parameter()); // param buffer in main()
107 const bool is_entry_parameter = in HloBufferIsReadOnly() local111 if (is_entry_parameter) { in HloBufferIsReadOnly()1120 const bool is_entry_parameter = in AssignSingleHloBuffer() local1125 if (is_entry_parameter) { in AssignSingleHloBuffer()
109 return buffer_info.is_entry_parameter(); in ExtractEntryParamBufferInfos()