| /external/tensorflow/tensorflow/core/common_runtime/ |
| executor.h | 94 ScopedStepContainer* step_container = nullptr; member in struct:tensorflow::Executor::Args
|
| direct_session.h | 190 ScopedStepContainer step_container; member in struct:tensorflow::DirectSession::RunState
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| graph_mgr.cc | 477 ScopedStepContainer* step_container = new ScopedStepContainer( local
483 [this, item, collector, cost_graph, step_container,
487 delete step_container;
495 args.step_container = step_container;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| captured_function.cc | 305 ScopedStepContainer step_container(
309 f_opts.step_container = &step_container;
341 ScopedStepContainer step_container(
345 f_opts.step_container = &step_container;
377 ScopedStepContainer step_container(
381 f_opts.step_container = &step_container;
421 ScopedStepContainer* step_container = new ScopedStepContainer local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| stack.cc | 144 auto* step_container = ctx->step_container(); local
145 if (step_container == nullptr) {
148 TF_RETURN_IF_ERROR(rm->Lookup(step_container->name(), key, stack));
188 auto* step_container = ctx->step_container(); local
189 OP_REQUIRES(ctx, step_container != nullptr,
191 OP_REQUIRES_OK(ctx, rm->Create(step_container->name(), key, stack));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_compiler.cc | 127 auto step_container = absl::make_unique<ScopedStepContainer>( local
132 step_container->name(), XlaContext::kXlaContextResourceName,
135 GraphCompiler graph_compiler(device, graph.get(), flib, step_container.get());
138 step_container.reset();
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| function.h | 620 ScopedStepContainer* step_container = nullptr; member in struct:tensorflow::FunctionLibraryRuntime::Options
[all...] |
| op_kernel.h | 627 ScopedStepContainer* step_container = nullptr; member in struct:tensorflow::OpKernelContext::Params
1163 ScopedStepContainer* step_container() const { function in class:tensorflow::OpKernelContext
[all...] |
