Searched refs:computation_proto (Results 1 – 2 of 2) sorted by relevance
65 const xrt::XLAComputation& computation_proto,83 const xrt::XLAComputation& computation_proto, in Compile() argument85 const xrt::XLAComputationConfig& config = computation_proto.config(); in Compile()112 client->LoadSnapshot(computation_proto.hlo_snapshot())); in Compile()151 xrt::XLAComputation computation_proto; in Compute() local154 computation_proto.ParseFromString(computation_input.scalar<string>()()), in Compute()159 OP_REQUIRES_OK(ctx, CompilationCacheKey(computation_proto, &key)); in Compute()177 return Compile(ctx, computation_proto, program); in Compute()
230 HloComputationProto computation_proto = computation->ToProto(); in ToProto() local231 proto.add_computations()->Swap(&computation_proto); in ToProto()309 for (const HloComputationProto& computation_proto : proto.computations()) { in CreateFromProto() local312 HloComputation::CreateFromProto(computation_proto, computation_map)); in CreateFromProto()314 int64 computation_id = computation_proto.id(); in CreateFromProto()