Searched defs:HloProtoEvaluator (Results 1 – 1 of 1) sorted by relevance
105 struct HloProtoEvaluator { struct106 explicit HloProtoEvaluator(HloEvaluator& evaluator, HloInstructionProto inst) in HloProtoEvaluator() argument113 HloProtoEvaluator& WithComputation( in WithComputation()126 HloProtoEvaluator& WithPrimitiveType(PrimitiveType new_primitive_type) { in WithPrimitiveType()132 HloProtoEvaluator& WithOpCode(HloOpcode new_opcode) { in WithOpCode()138 HloProtoEvaluator& WithOperands(absl::Span<Literal> operands) { in WithOperands()145 HloProtoEvaluator& WithSubshape(ShapeIndex shape_index) { in WithSubshape()150 StatusOr<Literal> Evaluate() { in Evaluate()195 HloEvaluator& evaluator;196 HloInstructionProto inst;[all …]