Searched refs:const_instr (Results 1 – 2 of 2) sorted by relevance
24 HloInstruction* const_instr = nullptr; in InstrForConstantBufferAllocation() local35 CHECK_EQ(const_instr, nullptr) in InstrForConstantBufferAllocation()36 << const_instr->ToString() << " " << buffer->ToString(); in InstrForConstantBufferAllocation()37 const_instr = buffer->instruction(); in InstrForConstantBufferAllocation()40 CHECK_NE(const_instr, nullptr); in InstrForConstantBufferAllocation()41 return *const_instr; in InstrForConstantBufferAllocation()
2581 HloInstructionProto const_instr; in BuildConstantSubGraph() local2582 *const_instr.mutable_shape() = literal.shape().ToProto(); in BuildConstantSubGraph()2583 *const_instr.mutable_literal() = literal.ToProto(); in BuildConstantSubGraph()2584 *const_instr.mutable_opcode() = HloOpcodeString(HloOpcode::kConstant); in BuildConstantSubGraph()2586 const_instr.set_id(handle); in BuildConstantSubGraph()2587 *const_instr.mutable_name() = in BuildConstantSubGraph()2588 GetFullName(const_instr.opcode(), kNameSeparator, const_instr.id()); in BuildConstantSubGraph()2590 const_instr; // Add to the result constant graph. in BuildConstantSubGraph()