| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 164 const SessionMetadata* session_metadata, in CompileLocallyAndFillHostCache() argument
170 flib_runtime, session_metadata, mesh_state, dynamic_shapes, in CompileLocallyAndFillHostCache()
179 const SessionMetadata* session_metadata, in CompileLocallyAndFillHostCacheInternal() argument
209 const std::string session_name = SessionNameFromMetadata(session_metadata); in CompileLocallyAndFillHostCacheInternal()
290 key, ctx->session_metadata(), ref_holder, &uid, &proto_key, &sharding_key, in ComputeInternal()
312 ctx->function_library(), ctx->session_metadata(), mesh_state, in ComputeInternal()
316 ctx->function_library(), ctx->session_metadata(), mesh_state, in ComputeInternal()
|
| 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_compile_op_common.h | 119 const SessionMetadata* session_metadata,
128 const SessionMetadata* session_metadata,
139 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 | 367 const SessionMetadata* session_metadata, in CompileIfKeyAbsent() argument
375 subgraph_key, session_metadata, per_step_ref_holder, uid, proto_key, in CompileIfKeyAbsent()
405 const SessionMetadata* session_metadata, in CompileIfKeyAbsentHelper() argument
427 tpu::SessionNameFromMetadata(session_metadata); in CompileIfKeyAbsentHelper()
|
| D | tpu_util.h | 36 std::string SessionNameFromMetadata(const SessionMetadata* session_metadata);
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_utils.cc | 45 params.session_metadata = &fallback_request_state.session_metadata(); in SetUpParams()
|
| D | kernel_fallback_compat_request_state.h | 138 const SessionMetadata& session_metadata() const { return session_metadata_; } in session_metadata() function
|
| D | kernel_fallback_execute_compat.cc | 89 fallback_request_state->session_metadata().name(), " (", in KernelFallbackEmitError()
90 fallback_request_state->session_metadata().version(), ") "); in KernelFallbackEmitError()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 92 SessionMetadata session_metadata; in GenerateSessionMetadata() local
93 session_metadata.set_name("name"); in GenerateSessionMetadata()
94 session_metadata.set_version(42); in GenerateSessionMetadata()
95 return session_metadata; in GenerateSessionMetadata()
135 const SessionMetadata* session_metadata = nullptr) { in Init() argument
144 /*thread_pool=*/nullptr, cluster_flr_.get(), session_metadata, in Init()
1184 if (ctx->session_metadata() != nullptr) { in Compute()
1185 out_tensor->scalar<tstring>()() = ctx->session_metadata()->DebugString(); in Compute()
1223 const SessionMetadata session_metadata = GenerateSessionMetadata(); in TEST_F() local
1224 Init({SessionMetadataReaderOpFn()}, &session_metadata); in TEST_F()
[all …]
|
| D | function.h | 64 const SessionMetadata* session_metadata,
|
| D | local_executor_params.h | 38 const SessionMetadata* session_metadata = nullptr; member
|
| D | direct_session.cc | 175 if (experimental_config.session_metadata().version() < 0) { in NewSession()
178 experimental_config.session_metadata().DebugString()); in NewSession()
180 const string key = GetMetadataKey(experimental_config.session_metadata()); in NewSession()
186 experimental_config.session_metadata().DebugString()); in NewSession()
236 session->options().config.experimental().session_metadata())); in Deregister()
522 options_.config.experimental().session_metadata(); in RunInternal()
548 options_.config.experimental().session_metadata(); in RunInternal()
1358 const auto* session_metadata = in CreateExecutors() local
1360 ? &options_.config.experimental().session_metadata() in CreateExecutors()
1365 /*parent=*/nullptr, session_metadata, in CreateExecutors()
[all …]
|
| D | direct_session_test.cc | 1170 if (ctx->session_metadata() != nullptr) { in Compute()
1171 out_tensor->scalar<tstring>()() = ctx->session_metadata()->DebugString(); in Compute()
1245 auto* session_metadata = in TEST() local
1247 session_metadata->set_name("name"); in TEST()
1248 session_metadata->set_version(1); in TEST()
1272 EXPECT_EQ(session_metadata->name(), metadata.session_metadata().name()); in TEST()
1273 EXPECT_EQ(session_metadata->version(), metadata.session_metadata().version()); in TEST()
1289 auto* session_metadata = in TEST() local
1291 session_metadata->set_name("name"); in TEST()
1292 session_metadata->set_version(1); in TEST()
[all …]
|
| D | function.cc | 333 const SessionMetadata* session_metadata,
471 const SessionMetadata* session_metadata, in FunctionLibraryRuntimeImpl() argument
480 session_metadata_(session_metadata), in FunctionLibraryRuntimeImpl()
965 params.session_metadata = session_metadata_; in CreateItem()
1414 const SessionMetadata* session_metadata, in NewFunctionLibraryRuntime() argument
1418 optimizer_options, session_metadata, parent)); in NewFunctionLibraryRuntime()
|
| D | single_threaded_executor.cc | 315 params.session_metadata = params_.session_metadata; in Run()
|
| D | process_function_library_runtime.h | 73 const SessionMetadata* session_metadata = nullptr,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-run-metadata.pbtxt | 34 name: "session_metadata"
|
| D | tensorflow.-config-proto.-experimental.pbtxt | 60 name: "session_metadata"
|
| D | tensorflow.-config-proto.pbtxt | 189 name: "session_metadata"
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_batch_tf_opkernels.cc | 499 const auto& session_metadata = src_fallback_request_state->session_metadata(); in SetUpKernelFallbackCompatRequestContextForBatch() local
507 builder, device_manager, pflr, intra_op_threadpool, session_metadata, in SetUpKernelFallbackCompatRequestContextForBatch()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/inference/ |
| D | batch_op_rewriter.cc | 133 config_proto_.experimental().session_metadata().name(); in Optimize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 92 if (!ctx->session_metadata()) return *kModelNameUnset; in GetModelName()
93 if (ctx->session_metadata()->name().empty()) return *kModelNameUnset; in GetModelName()
94 return ctx->session_metadata()->name(); in GetModelName()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 203 if (!ctx->session_metadata()) return *kModelNameUnset; in GetModelName()
204 if (ctx->session_metadata()->name().empty()) return *kModelNameUnset; in GetModelName()
205 return ctx->session_metadata()->name(); in GetModelName()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 634 const SessionMetadata* session_metadata = nullptr; member
1072 const SessionMetadata* session_metadata() const { in session_metadata() function
1073 return params_->session_metadata; in session_metadata()
|