| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_util.cc | 25 std::string SessionNameFromMetadata(const SessionMetadata* session_metadata) { in SessionNameFromMetadata() argument
26 return session_metadata ? session_metadata->name() : ""; in SessionNameFromMetadata()
|
| D | tpu_util.h | 36 std::string SessionNameFromMetadata(const SessionMetadata* session_metadata);
|
| D | tpu_compile_op_common.h | 120 const SessionMetadata* session_metadata,
131 const SessionMetadata* session_metadata, in LookupPersistentCompilationCacheAndFillCaches() argument
|
| D | tpu_compilation_cache_interface.h | 109 const SessionMetadata* session_metadata,
198 const SessionMetadata* session_metadata,
|
| D | tpu_compilation_cache_interface.cc | 363 const SessionMetadata* session_metadata, in CompileIfKeyAbsent() argument
371 subgraph_key, session_metadata, per_step_ref_holder, uid, proto_key, in CompileIfKeyAbsent()
401 const SessionMetadata* session_metadata, in CompileIfKeyAbsentHelper() argument
423 tpu::SessionNameFromMetadata(session_metadata); in CompileIfKeyAbsentHelper()
|
| D | tpu_compile_op_common.cc | 592 const SessionMetadata* session_metadata, in CompileLocallyAndFillHostCache() argument
617 const std::string session_name = SessionNameFromMetadata(session_metadata); in CompileLocallyAndFillHostCache()
693 key, ctx->session_metadata(), ref_holder, &uid, &proto_key, &sharding_key, in ComputeInternal()
715 ctx->function_library(), ctx->session_metadata(), mesh_state, in ComputeInternal()
719 ctx->function_library(), ctx->session_metadata(), mesh_state, in ComputeInternal()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 90 SessionMetadata session_metadata; in GenerateSessionMetadata() local
91 session_metadata.set_name("name"); in GenerateSessionMetadata()
92 session_metadata.set_version(42); in GenerateSessionMetadata()
93 return session_metadata; in GenerateSessionMetadata()
133 const SessionMetadata* session_metadata = nullptr) { in Init() argument
142 /*thread_pool=*/nullptr, cluster_flr_.get(), session_metadata, in Init()
1135 if (ctx->session_metadata() != nullptr) { in Compute()
1136 out_tensor->scalar<tstring>()() = ctx->session_metadata()->DebugString(); in Compute()
1174 const SessionMetadata session_metadata = GenerateSessionMetadata(); in TEST_F() local
1175 Init({SessionMetadataReaderOpFn()}, &session_metadata); in TEST_F()
[all …]
|
| D | local_executor_params.h | 38 const SessionMetadata* session_metadata = nullptr; member
|
| D | function.h | 64 const SessionMetadata* session_metadata,
|
| D | direct_session.cc | 170 if (experimental_config.session_metadata().version() < 0) { in NewSession()
173 experimental_config.session_metadata().DebugString()); in NewSession()
175 const string key = GetMetadataKey(experimental_config.session_metadata()); in NewSession()
181 experimental_config.session_metadata().DebugString()); in NewSession()
231 session->options().config.experimental().session_metadata())); in Deregister()
510 options_.config.experimental().session_metadata(); in RunInternal()
1341 const auto* session_metadata = in CreateExecutors() local
1343 ? &options_.config.experimental().session_metadata() in CreateExecutors()
1348 /*parent=*/nullptr, session_metadata, in CreateExecutors()
1373 params.session_metadata = session_metadata; in CreateExecutors()
|
| D | function.cc | 329 const SessionMetadata* session_metadata,
463 const SessionMetadata* session_metadata, in FunctionLibraryRuntimeImpl() argument
472 session_metadata_(session_metadata), in FunctionLibraryRuntimeImpl()
945 params.session_metadata = session_metadata_; in CreateItem()
1377 const SessionMetadata* session_metadata, in NewFunctionLibraryRuntime() argument
1381 optimizer_options, session_metadata, parent)); in NewFunctionLibraryRuntime()
|
| D | direct_session_test.cc | 1169 if (ctx->session_metadata() != nullptr) { in Compute()
1170 out_tensor->scalar<tstring>()() = ctx->session_metadata()->DebugString(); in Compute()
1244 auto* session_metadata = in TEST() local
1246 session_metadata->set_name("name"); in TEST()
1247 session_metadata->set_version(1); in TEST()
1275 auto* session_metadata = in TEST() local
1277 session_metadata->set_name("name"); in TEST()
1278 session_metadata->set_version(1); in TEST()
|
| D | process_function_library_runtime.h | 72 const SessionMetadata* session_metadata = nullptr,
|
| D | executor.cc | 401 session_metadata_(immutable_state.params().session_metadata), in ExecutorState()
702 params.session_metadata = session_metadata_; in Process()
|
| D | process_function_library_runtime.cc | 91 const SessionMetadata* session_metadata, in ProcessFunctionLibraryRuntime() argument
102 session_metadata_(session_metadata), in ProcessFunctionLibraryRuntime()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-config-proto.-experimental.pbtxt | 60 name: "session_metadata"
|
| D | tensorflow.-config-proto.pbtxt | 189 name: "session_metadata"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 75 if (!ctx->session_metadata()) return *kModelNameUnset; in GetModelName()
76 if (ctx->session_metadata()->name().empty()) return *kModelNameUnset; in GetModelName()
77 return ctx->session_metadata()->name(); in GetModelName()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 123 if (!ctx->session_metadata()) return *kModelNameUnset; in GetModelName()
124 if (ctx->session_metadata()->name().empty()) return *kModelNameUnset; in GetModelName()
125 return ctx->session_metadata()->name(); in GetModelName()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 661 const SessionMetadata* session_metadata = nullptr; member
1116 const SessionMetadata* session_metadata() const { in session_metadata() function
1117 return params_->session_metadata; in session_metadata()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 561 SessionMetadata session_metadata = 11; field
|