| /external/tensorflow/tensorflow/core/grappler/costs/ |
| op_context.h | 31 const FunctionDefLibrary* function_library; // Not owned. member in struct:tensorflow::grappler::OpContext
33 OpContext() { function_library = nullptr; }
|
| graph_properties.cc | 924 FunctionLibraryDefinition function_library(graph.op_registry(),
929 shape_refiner.set_function_library_for_shape_inference(&function_library);
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| graph_runner.cc | 104 Status GraphRunner::Run(Graph* graph, FunctionLibraryRuntime* function_library,
112 if (function_library && function_library->device() &&
113 function_library->device()->device_type() != cpu_device_->device_type()) {
115 // so just ignore the function_library.
117 // identical to function_library except that it uses CPU?
119 << function_library->device()->device_type() << " device.";
120 function_library = nullptr;
162 params.function_library = function_library;
[all...] |
| graph_runner.h | 54 // `function_library` may be nullptr.
56 Status Run(Graph* graph, FunctionLibraryRuntime* function_library,
|
| function_testlib.cc | 36 ctx->function_library()->device()->name();
|
| constant_folding.h | 62 FunctionLibraryRuntime* function_library, Env* env,
|
| kernel_benchmark_testlib.cc | 77 params.function_library = nullptr;
|
| executor.h | 134 FunctionLibraryRuntime* function_library = nullptr; member in struct:tensorflow::LocalExecutorParams
|
| constant_folding.cc | 554 FunctionLibraryRuntime* function_library, Env* env,
619 graph_runner->Run(constant_graph.get(), function_library, {} /* inputs*/,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| critical_section.cc | 175 auto* function_library = ctx_->function_library(); local
176 s = function_library->Instantiate(func_->name(), AttrSlice(&func_->attr()),
192 function_library->Run(opts_, handle_, arguments_t_, &output_t_,
|
| functional_ops.cc | 114 auto lib = ctx->function_library();
143 lib_(CHECK_NOTNULL(ctx_->function_library())) {
196 auto lib = ctx->function_library();
244 lib_(CHECK_NOTNULL(ctx_->function_library())) {
|
| function_ops.cc | 234 FunctionLibraryRuntime* lib = ctx->function_library();
301 FunctionLibraryRuntime* lib = ctx->function_library();
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| iterator_ops.cc | 147 *ctx->function_library()->GetFunctionLibraryDefinition()));
161 params.function_library = flib_def;
181 std::shared_ptr<const FunctionLibraryDefinition> function_library() { function in class:tensorflow::__anon39744::IteratorResource
477 OP_REQUIRES_OK(context, context->function_library()->Clone(
554 *ctx->function_library()->GetFunctionLibraryDefinition()));
611 params.lib = ctx->function_library();
612 DeviceBase* device = ctx->function_library()->device();
741 TF_RETURN_IF_ERROR(ctx->function_library()->Clone(&flib_def, &pflr, &lib));
765 TF_RETURN_IF_ERROR(ctx->function_library()->Instantiate(
784 ctx->function_library()->Run(opts, f_handle, {}, &return_values
[all...] |
| flat_map_dataset_op.cc | 252 params.lib = ctx->function_library();
|
| captured_function.cc | 169 inst_opts.overlay_lib = ctx->function_library().get();
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| function_ops.cc | 66 FunctionLibraryRuntime* lib = ctx->function_library();
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_op_kernel.h | 190 FunctionLibraryRuntime* function_library() const { function in class:tensorflow::XlaOpKernelContext
191 return context_->function_library();
|
| /external/tensorflow/tensorflow/core/framework/ |
| dataset.h | 172 ctx->function_library()->GetFunctionLibraryDefinition();
274 std::shared_ptr<const FunctionLibraryDefinition> function_library = nullptr; member in struct:tensorflow::IteratorContext::Params
296 std::shared_ptr<const FunctionLibraryDefinition> function_library() { function in class:tensorflow::IteratorContext
297 return params_.function_library;
|
| op_kernel.h | 320 // CHECK_NOTNULL(function_library())->Instantiate("Foo", ...).
321 FunctionLibraryRuntime* function_library() const { return flib_; } function in class:tensorflow::OpKernelConstruction
562 FunctionLibraryRuntime* function_library = nullptr; member in struct:tensorflow::OpKernelContext::Params
951 FunctionLibraryRuntime* function_library() const { function in class:tensorflow::OpKernelContext
[all...] |
| dataset.cc | 145 ctx->function_library()->GetFunctionLibraryDefinition();
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| xla_launch_op.cc | 258 options.flib_def = ctx->function_library()->GetFunctionLibraryDefinition();
259 options.graph_def_version = ctx->function_library()->graph_def_version();
|
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder.cc | 103 FunctionLibraryDefinition function_library(OpRegistry::Global(),
121 &function_library, *optimizer_opts));
128 std::unique_ptr<Graph> graphptr(new Graph(function_library));
|
| /external/tensorflow/tensorflow/c/eager/ |
| runtime.cc | 284 params.function_library = flib_;
|
| /external/tensorflow/tensorflow/contrib/data/kernels/ |
| prefetching_kernels.cc | 264 FunctionLibraryRuntime* lib = ctx->function_library();
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| graph_mgr.cc | 228 params.function_library = lib;
|
