| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | versioned_computation_handle.cc | 27 const VersionedComputationHandle& versioned_handle) {
28 out << versioned_handle.ToString();
26 operator <<(std::ostream& out, const VersionedComputationHandle& versioned_handle) argument
|
| H A D | compilation_cache.cc | 51 const VersionedComputationHandle& versioned_handle,
55 CacheKey key = BuildKey(versioned_handle, module_config);
69 const VersionedComputationHandle& versioned_handle,
73 return tensorflow::strings::StrCat(versioned_handle.handle.handle(), "::",
74 versioned_handle.version, "::",
50 LookUp( const VersionedComputationHandle& versioned_handle, const HloModuleConfig& module_config) const argument
68 BuildKey( const VersionedComputationHandle& versioned_handle, const HloModuleConfig& module_config) const argument
|
| H A D | compilation_cache.h | 58 const VersionedComputationHandle& versioned_handle,
68 CacheKey BuildKey(const VersionedComputationHandle& versioned_handle,
|
| H A D | computation_tracker.cc | 148 const VersionedComputationHandle& versioned_handle,
151 if (visited->count(versioned_handle) > 0) {
152 CHECK_EQ(1, visited->count(versioned_handle));
157 ResolveInternal(versioned_handle.handle).ValueOrDie();
159 computation->GetEmbeddedComputations(versioned_handle.version);
165 visited->insert(versioned_handle);
166 post_order->push_back(versioned_handle);
196 const VersionedComputationHandle& versioned_handle) -> HloComputation* {
197 CHECK_GT(hlo_computations.count(versioned_handle), 0);
198 return hlo_computations.at(versioned_handle);
147 ComputeComputationPostOrder( const VersionedComputationHandle& versioned_handle, std::set<VersionedComputationHandle>* visited, std::list<VersionedComputationHandle>* post_order) const argument
[all...] |
| H A D | compile_only_service.cc | 72 VersionedComputationHandle versioned_handle = local
83 computation_tracker_.SnapshotComputation(versioned_handle.handle));
85 "computation_", versioned_handle.handle.handle(), "__",
87 versioned_handle.version);
97 user_computation->ComputeProgramShape(versioned_handle.version));
108 versioned_handle, *module_config,
|
| H A D | versioned_computation_handle.h | 51 const VersionedComputationHandle& versioned_handle);
|
| H A D | service.cc | 370 for (const VersionedComputationHandle& versioned_handle : versioned_handles) {
371 VLOG(1) << versioned_handle;
377 const VersionedComputationHandle& versioned_handle = versioned_handles[i]; local
381 versioned_handle, config,
401 const VersionedComputationHandle& versioned_handle,
405 versioned_handle.ToString().c_str());
416 computation_tracker_.SnapshotComputation(versioned_handle.handle));
419 versioned_handle.handle.handle(),
421 versioned_handle.version);
429 computation_tracker_.BuildHloModule(versioned_handle, *module_confi
400 BuildExecutable( const VersionedComputationHandle& versioned_handle, std::unique_ptr<HloModuleConfig> module_config, Backend* backend, se::StreamExecutor* executor, DeviceMemoryAllocator* device_allocator) argument
450 BuildAndCacheExecutable( const VersionedComputationHandle& versioned_handle, std::unique_ptr<HloModuleConfig> module_config, Backend* backend, perftools::gputools::StreamExecutor* executor, ExecutionProfile* profile, DeviceMemoryAllocator* device_allocator) argument
735 VersionedComputationHandle versioned_handle = local
842 VersionedComputationHandle versioned_handle = local
919 VersionedComputationHandle versioned_handle = local
1183 VersionedComputationHandle versioned_handle = local
1203 VersionedComputationHandle versioned_handle = local
1301 VersionedComputationHandle versioned_handle = local
1325 VersionedComputationHandle versioned_handle = local
[all...] |
| H A D | local_service.cc | 78 VersionedComputationHandle versioned_handle = local
83 user_computation->ComputeProgramShape(versioned_handle.version));
144 return BuildExecutable(versioned_handle, std::move(module_config),
|
| H A D | computation_tracker.h | 121 const VersionedComputationHandle& versioned_handle,
|
| H A D | executable.cc | 132 VersionedComputationHandle versioned_handle = entry_computation_handle(); local
136 "computation_%lld__%s__execution_%lld", versioned_handle.handle.handle(),
|
| H A D | service.h | 288 const VersionedComputationHandle& versioned_handle,
306 const VersionedComputationHandle& versioned_handle,
|
| H A D | hlo_cost_analysis_test.cc | 128 VersionedComputationHandle versioned_handle = local
131 computation_tracker_.BuildHloModule(versioned_handle, HloModuleConfig())
|
| H A D | user_computation.cc | 1527 VersionedComputationHandle versioned_handle; local
1528 versioned_handle.handle = session_computation_.computation_handle();
1532 versioned_handle.version = handle_to_return_.handle();
1537 versioned_handle.version = next_handle_value_ - 1;
1540 return versioned_handle;
1549 VersionedComputationHandle versioned_handle; local
1550 versioned_handle.handle = session_computation_.computation_handle();
1551 versioned_handle.version = operation.handle();
1552 return versioned_handle;
2121 const VersionedComputationHandle versioned_handle local
2131 const VersionedComputationHandle versioned_handle = { local
2141 const VersionedComputationHandle versioned_handle = { local
2152 const VersionedComputationHandle versioned_handle = { local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | dumped_computation_to_text.cc | 80 VersionedComputationHandle versioned_handle = local
83 tracker.BuildHloModule(versioned_handle, HloModuleConfig())
|