Searched refs:const_instr (Results 1 – 3 of 3) 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()
897 auto const_instr = xla::Cast<xla::HloConstantInstruction>(instr); in EmitConstant() local898 TF_RET_CHECK(const_instr->shape().IsArray() && in EmitConstant()899 const_instr->shape().is_static()); in EmitConstant()901 const_instr->shape(), builder_)); in EmitConstant()907 CreateDenseElementsAttrFromLiteral(const_instr->literal(), builder_)); in EmitConstant()
3798 HloInstructionProto const_instr; in BuildConstantSubGraph() local3819 *const_instr.mutable_literal() = literal.ToProto(); in BuildConstantSubGraph()3820 *const_instr.mutable_shape() = literal.shape().ToProto(); in BuildConstantSubGraph()3822 *const_instr.mutable_literal() = instr_proto->literal(); in BuildConstantSubGraph()3823 *const_instr.mutable_shape() = instr_proto->shape(); in BuildConstantSubGraph()3825 *const_instr.mutable_opcode() = HloOpcodeString(HloOpcode::kConstant); in BuildConstantSubGraph()3826 const_instr.set_id(handle); in BuildConstantSubGraph()3827 *const_instr.mutable_name() = in BuildConstantSubGraph()3828 GetFullName(const_instr.opcode(), kNameSeparator, const_instr.id()); in BuildConstantSubGraph()3830 const_instr; // Add to the result constant graph. in BuildConstantSubGraph()