Home
last modified time | relevance | path

Searched refs:graph_def_version (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dcpp_shape_inference.cc52 int graph_def_version, const string& serialized_node_def, in RunCppShapeInferenceImpl() argument
128 graph_def_version, &node, op_reg_data->op_def, input_shapes, in RunCppShapeInferenceImpl()
174 int graph_def_version, const string& serialized_node_def, in RunCppShapeInference() argument
195 graph_def_version, serialized_node_def, input_serialized_shapes, in RunCppShapeInference()
Dcpp_shape_inference.h45 int graph_def_version, const string& serialized_node_def,
Dtest_ops.cc48 if (c->graph_def_version() != TF_GRAPH_DEF_VERSION - 1) { in __anonddbb40510202()
133 : OpKernel(ctx), graph_def_version_(ctx->graph_def_version()) {} in GraphDefVersionOp()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h196 int graph_def_version, bool is_stateful,
263 int graph_def_version,
273 int graph_def_version,
Dfunctions.cc314 std::vector<ControlOutput> control_outputs, const int graph_def_version, in GrapplerFunctionItem() argument
325 graph.mutable_versions()->set_producer(graph_def_version); in GrapplerFunctionItem()
484 const int graph_def_version, in MakeGrapplerFunctionItem() argument
651 std::move(outputs), std::move(control_outputs), graph_def_version, in MakeGrapplerFunctionItem()
658 const int graph_def_version, in MakeGrapplerFunctionItem() argument
660 return MakeGrapplerFunctionItem(func, AttrSlice(), flib, graph_def_version, in MakeGrapplerFunctionItem()
/external/tensorflow/tensorflow/core/common_runtime/
Dkernel_benchmark_testlib.cc75 const int graph_def_version = g->versions().producer(); in Benchmark() local
80 params.create_kernel = [this, graph_def_version](const NodeDef& ndef, in Benchmark()
83 graph_def_version, kernel); in Benchmark()
Dfunction.h58 int graph_def_version, const FunctionLibraryDefinition* lib_def,
68 int graph_def_version, const FunctionLibraryDefinition* lib_def,
Dprocess_function_library_runtime.h35 const DeviceMgr* device_mgr, Env* env, int graph_def_version,
43 int graph_def_version,
216 Status Clone(Env* env, int graph_def_version,
Dprocess_function_library_runtime.cc59 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime() argument
72 nullptr, env, nullptr, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
78 device_mgr, env, d, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
84 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime() argument
98 nullptr, env, nullptr, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
104 device_mgr, env, d, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
1004 Env* env, int graph_def_version, const OptimizerOptions& optimizer_options, in Clone() argument
1010 device_mgr_, env, graph_def_version, out_lib_def->get(), in Clone()
Dshape_refiner.h91 ShapeRefiner(int graph_def_version, const OpRegistryInterface* ops);
147 int32 graph_def_version() const { return graph_def_version_; } in graph_def_version() function
Deval_const_tensor.h58 const OpRegistryInterface& ops, int32 graph_def_version, bool* evaluated,
Dfunction.cc201 int graph_def_version() override;
291 int FunctionLibraryRuntimeOverlay::graph_def_version() { in graph_def_version() function in tensorflow::FunctionLibraryRuntimeOverlay
292 return base_flr_->graph_def_version(); in graph_def_version()
308 int graph_def_version,
351 int graph_def_version() override { return graph_def_version_; } in graph_def_version() function in tensorflow::FunctionLibraryRuntimeImpl
425 const DeviceMgr* dmgr, Env* env, Device* device, int graph_def_version, in FunctionLibraryRuntimeImpl() argument
434 graph_def_version_(graph_def_version), in FunctionLibraryRuntimeImpl()
1221 int graph_def_version, const FunctionLibraryDefinition* lib_def, in NewFunctionLibraryRuntime() argument
1226 device_mgr, env, device, graph_def_version, lib_def, thread_pool, in NewFunctionLibraryRuntime()
1232 int graph_def_version, const FunctionLibraryDefinition* lib_def, in NewFunctionLibraryRuntime() argument
[all …]
Deval_const_tensor.cc310 int32 graph_def_version, bool* evaluated, in EvaluateConstantTensor() argument
333 versions.set_producer(graph_def_version); in EvaluateConstantTensor()
Dexecutor.h226 const NodeDef& ndef, int graph_def_version,
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc105 graph_def_version_(context->graph_def_version()), in OpKernel()
114 context->graph_def_version())); in OpKernel()
206 const MemoryTypeSlice& output_memory_types, int graph_def_version, in OpKernelConstruction() argument
218 graph_def_version_(graph_def_version), in OpKernelConstruction()
1258 const NodeDef& node_def, int graph_def_version, Status* status) { in CreateOpKernel() argument
1261 node_def, graph_def_version, &kernel); in CreateOpKernel()
1267 const NodeDef& node_def, int graph_def_version, in CreateOpKernel() argument
1325 input_memory_types, outputs, output_memory_types, graph_def_version, &s); in CreateOpKernel()
Dshape_inference.h173 InferenceContext(int graph_def_version, const NodeDef* node_def,
193 int graph_def_version, const NodeDef* node_def, const OpDef& op_def,
213 int graph_def_version, const NodeDef* node_def, const OpDef& op_def,
638 int graph_def_version() const { return graph_def_version_; } in graph_def_version() function
Dshape_inference_testutil.h41 int graph_def_version = TF_GRAPH_DEF_VERSION; member
Dop_def_util.h35 Status CheckOpDeprecation(const OpDef& op_def, int graph_def_version);
Dshape_inference.cc33 int graph_def_version, const NodeDef* node_def, const OpDef& op_def, in InferenceContext() argument
40 : graph_def_version_(graph_def_version), in InferenceContext()
88 int graph_def_version, const NodeDef* node_def, const OpDef& op_def, in InferenceContext() argument
95 : graph_def_version_(graph_def_version), in InferenceContext()
141 int graph_def_version, const NodeDef* node_def, const OpDef& op_def, in InferenceContext() argument
147 : graph_def_version_(graph_def_version), in InferenceContext()
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h102 Status InitOpWithGraphVersion(int graph_def_version) { in InitOpWithGraphVersion() argument
105 node_def_, graph_def_version, &status); in InitOpWithGraphVersion()
/external/tensorflow/tensorflow/core/kernels/data/
Doptimize_dataset_op.cc36 graph_def_version_(ctx->graph_def_version()) { in OptimizeDatasetOp()
Dfilter_by_component_dataset_op.cc36 graph_def_version_(ctx->graph_def_version()) { in FilterByLastComponentDatasetOp()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drebatch_dataset_op.cc28 graph_def_version_(ctx->graph_def_version()) { in RebatchDatasetOp()
Dauto_shard_dataset_op.cc28 graph_def_version_(ctx->graph_def_version()) { in AutoShardDatasetOp()
/external/tensorflow/tensorflow/compiler/jit/
Dcreate_xla_launch_op.cc235 fbody->ret_types, output_memory_types, flr->graph_def_version(), &s); in CreateXlaLaunchOp()

12