Searched refs:LookUpInstruction (Results 1 – 2 of 2) sorted by relevance
107 TF_ASSIGN_OR_RETURN(auto instr, LookUpInstruction(op)); in GetShape()2511 TF_RETURN_IF_ERROR(LookUpInstruction(operand).status()); in IsConstant()2523 auto op_status = LookUpInstruction(root_op); in BuildConstantSubGraph()2539 LookUpInstruction(root_op)); in BuildConstantSubGraph()2798 StatusOr<const HloInstructionProto*> XlaBuilder::LookUpInstruction( in LookUpInstruction() function in xla::XlaBuilder
591 StatusOr<const HloInstructionProto*> LookUpInstruction(const XlaOp& op) const;