| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.h | 103 int graph_def_version, bool is_stateful,
164 int graph_def_version,
174 int graph_def_version,
|
| D | functions.cc | 44 std::vector<ControlOutput> control_outputs, const int graph_def_version, in GrapplerFunctionItem() argument
55 graph.mutable_versions()->set_producer(graph_def_version); in GrapplerFunctionItem()
224 const int graph_def_version, in MakeGrapplerFunctionItem() argument
298 graph_def_version, signature.is_stateful(), std::move(function_body)); in MakeGrapplerFunctionItem()
304 const int graph_def_version, in MakeGrapplerFunctionItem() argument
306 return MakeGrapplerFunctionItem(func, AttrSlice(), flib, graph_def_version, in MakeGrapplerFunctionItem()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | kernel_benchmark_testlib.cc | 88 const int graph_def_version = g->versions().producer(); in Benchmark() local
94 device_mgr_.get(), Env::Default(), nullptr, graph_def_version, in Benchmark()
103 params.create_kernel = [this, graph_def_version]( in Benchmark()
106 return CreateNonCachedKernel(device_, flr_, props, graph_def_version, in Benchmark()
|
| D | function.h | 61 Device* device, int graph_def_version,
|
| D | eval_const_tensor.h | 61 const OpRegistryInterface& ops, int32 graph_def_version, bool* evaluated,
|
| D | shape_refiner.h | 75 ShapeRefiner(int graph_def_version, const OpRegistryInterface* ops);
131 int32 graph_def_version() const { return graph_def_version_; } in graph_def_version() function
|
| D | function.cc | 191 int graph_def_version() const override;
307 int FunctionLibraryRuntimeOverlay::graph_def_version() const { in graph_def_version() function in tensorflow::FunctionLibraryRuntimeOverlay
308 return base_flr_->graph_def_version(); in graph_def_version()
325 int graph_def_version,
373 int graph_def_version() const override { return graph_def_version_; } in graph_def_version() function in tensorflow::FunctionLibraryRuntimeImpl
460 int graph_def_version, const FunctionLibraryDefinition* lib_def, in FunctionLibraryRuntimeImpl() argument
469 graph_def_version_(graph_def_version), in FunctionLibraryRuntimeImpl()
1374 Device* device, int graph_def_version, in NewFunctionLibraryRuntime() argument
1380 device_mgr, env, config, device, graph_def_version, lib_def, thread_pool, in NewFunctionLibraryRuntime()
|
| D | eval_const_tensor.cc | 350 int32 graph_def_version, bool* evaluated, in EvaluateConstantTensor() argument
390 versions.set_producer(graph_def_version); in EvaluateConstantTensor()
|
| D | executor.h | 238 int graph_def_version, OpKernel** kernel);
|
| D | process_function_library_runtime.h | 68 int graph_def_version, const FunctionLibraryDefinition* lib_def,
357 Status Clone(Env* env, int graph_def_version,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.cc | 142 options.graph_def_version = function_library.graph_def_version(); in GenerateCompilerOptions()
|
| D | xla_kernel_creator.cc | 164 flr->graph_def_version(), &s); in CreateXlaKernel()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 109 graph_def_version_(context->graph_def_version()), in OpKernel()
115 context->graph_def_version())); in OpKernel()
135 graph_def_version_(context->graph_def_version()), in OpKernel()
141 context->graph_def_version())); in OpKernel()
233 const MemoryTypeSlice& output_memory_types, int graph_def_version, in OpKernelConstruction() argument
243 graph_def_version_(graph_def_version), in OpKernelConstruction()
1578 const NodeDef& node_def, int graph_def_version, Status* status) { in CreateOpKernel() argument
1589 graph_def_version, status); in CreateOpKernel()
1594 const std::shared_ptr<const NodeProperties>& props, int graph_def_version, in CreateOpKernel() argument
1598 /*flib=*/nullptr, props, graph_def_version, &kernel); in CreateOpKernel()
[all …]
|
| D | shape_inference_testutil.h | 42 int graph_def_version = TF_GRAPH_DEF_VERSION; member
|
| D | op_def_util.h | 35 Status CheckOpDeprecation(const OpDef& op_def, int graph_def_version);
|
| D | op_kernel.h | 265 int graph_def_version, Status* status);
355 int graph_def_version() const { return graph_def_version_; } in graph_def_version() function
1379 int graph_def_version, Status* status);
1383 const std::shared_ptr<const NodeProperties>& props, int graph_def_version,
1389 int graph_def_version, OpKernel** kernel);
1395 int graph_def_version, OpKernel** kernel);
|
| D | shape_inference.h | 169 InferenceContext(int graph_def_version, const AttrSlice& attrs,
186 int graph_def_version, const AttrSlice& attrs, const OpDef& op_def,
628 int graph_def_version() const { return graph_def_version_; } in graph_def_version() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 141 Status OpsTestBase::InitOpWithGraphVersion(int graph_def_version) { in InitOpWithGraphVersion() argument
148 device_->resource_manager(), props, graph_def_version, &kernel)); in InitOpWithGraphVersion()
|
| D | ops_testutil.h | 91 Status InitOpWithGraphVersion(int graph_def_version);
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | functional_ops_test.cc | 39 op.graph_def_version, op.node_def, op_reg_data->op_def, in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.cc | 74 compiler_options.graph_def_version = graph->versions().producer(); in ConvertGraphToXla()
|
| D | xla_compiler.h | 161 int graph_def_version = TF_GRAPH_DEF_VERSION; member
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile.proto | 58 int32 graph_def_version = 10; field
|
| D | tpu_compile_op_support.h | 61 int graph_def_version; member
|
| D | tpu_compile_op_common.cc | 279 const FunctionLibraryDefinition& flib_def, int graph_def_version, in CompileTFFunctionToHlo() argument
294 compiler_options.graph_def_version = graph_def_version; in CompileTFFunctionToHlo()
609 flib_runtime->graph_def_version(), in CompileLocallyAndFillHostCache()
|