Searched refs:AotXlaComputationInstance (Results 1 – 6 of 6) sorted by relevance
26 const absl::Span<const AotXlaComputationInstance> computations, in CompileAheadOfTime()29 std::vector<CompileOnlyService::AotXlaComputationInstance> service_instances; in CompileAheadOfTime()31 for (const AotXlaComputationInstance& instance : computations) { in CompileAheadOfTime()33 CompileOnlyService::AotXlaComputationInstance& service_instance = in CompileAheadOfTime()
41 struct AotXlaComputationInstance { struct55 const absl::Span<const AotXlaComputationInstance> computations, argument
42 struct AotXlaComputationInstance { struct53 const absl::Span<const AotXlaComputationInstance> computations, argument58 const absl::Span<const AotXlaComputationInstance> computations,
66 const absl::Span<const AotXlaComputationInstance> computations, in CompileAheadOfTime()70 for (const AotXlaComputationInstance& instance : computations) { in CompileAheadOfTime()
80 xla::CompileOnlyClient::AotXlaComputationInstance instance{ in main()
71 xla::CompileOnlyClient::AotXlaComputationInstance instance; in CompileXla()