Searched refs:graph_def_version (Results 1 - 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
H A Dcpp_shape_inference.h45 int graph_def_version, const string& serialized_node_def,
H A Dcpp_shape_inference.cc52 int graph_def_version, const string& serialized_node_def,
128 graph_def_version, &node, op_reg_data->op_def, input_shapes,
174 int graph_def_version, const string& serialized_node_def,
195 graph_def_version, serialized_node_def, input_serialized_shapes,
51 RunCppShapeInferenceImpl( int graph_def_version, const string& serialized_node_def, const std::vector<string>& input_serialized_shapes, const std::vector<PyObject*>& input_constant_tensor_values, const std::vector<string>& input_constant_tensor_as_shape_values, std::vector<string>* output_tensor_shape_protos, string* input_tensors_needed_out) argument
173 RunCppShapeInference( int graph_def_version, const string& serialized_node_def, const std::vector<string>& input_serialized_shapes, PyObject* input_constant_tensor_values, const std::vector<string>& input_constant_tensor_as_shape_values, TF_Status* out_status) argument
H A Dtest_ops.cc48 if (c->graph_def_version() != TF_GRAPH_DEF_VERSION - 1) {
133 : OpKernel(ctx), graph_def_version_(ctx->graph_def_version()) {}
H A Dcommon_shapes.py653 graph_def_version = op.graph.graph_def_versions.producer
685 graph_def_version, node_def_str, input_shapes, input_tensors,
/external/tensorflow/tensorflow/core/common_runtime/
H A Dfunction.h57 int graph_def_version, const FunctionLibraryDefinition* lib_def,
67 int graph_def_version, const FunctionLibraryDefinition* lib_def,
H A Dprocess_function_library_runtime.h33 const DeviceMgr* device_mgr, Env* env, int graph_def_version,
40 int graph_def_version,
138 Status Clone(Env* env, int graph_def_version,
H A Dprocess_function_library_runtime.cc29 const DeviceMgr* device_mgr, Env* env, int graph_def_version,
39 NewFunctionLibraryRuntime(nullptr, env, nullptr, graph_def_version,
45 NewFunctionLibraryRuntime(device_mgr, env, d, graph_def_version,
51 const DeviceMgr* device_mgr, Env* env, int graph_def_version,
62 nullptr, env, nullptr, graph_def_version, lib_def, optimizer_options,
68 device_mgr, env, d, graph_def_version, lib_def, optimizer_options,
337 Env* env, int graph_def_version, const OptimizerOptions& optimizer_options,
343 device_mgr_, env, graph_def_version, out_lib_def->get(),
28 ProcessFunctionLibraryRuntime( const DeviceMgr* device_mgr, Env* env, int graph_def_version, const FunctionLibraryDefinition* lib_def, const OptimizerOptions& optimizer_options, DistributedFunctionLibraryRuntime* parent) argument
50 ProcessFunctionLibraryRuntime( const DeviceMgr* device_mgr, Env* env, int graph_def_version, const FunctionLibraryDefinition* lib_def, const OptimizerOptions& optimizer_options, CustomKernelCreator custom_kernel_creator, DistributedFunctionLibraryRuntime* parent) argument
336 Clone( Env* env, int graph_def_version, const OptimizerOptions& optimizer_options, CustomKernelCreator custom_kernel_creator, std::unique_ptr<FunctionLibraryDefinition>* out_lib_def, std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr) argument
H A Dkernel_benchmark_testlib.cc73 const int graph_def_version = g->versions().producer(); local
78 params.create_kernel = [this, graph_def_version](const NodeDef& ndef,
80 return CreateNonCachedKernel(device_, nullptr, ndef, graph_def_version,
H A Dshape_refiner.h91 ShapeRefiner(int graph_def_version, const OpRegistryInterface* ops);
147 int32 graph_def_version() const { return graph_def_version_; } function in class:tensorflow::ShapeRefiner
H A Dexecutor.h232 const NodeDef& ndef, int graph_def_version,
H A Dfunction.cc145 int graph_def_version,
181 int graph_def_version() override { return graph_def_version_; }
246 const DeviceMgr* dmgr, Env* env, Device* device, int graph_def_version,
254 graph_def_version_(graph_def_version),
946 int graph_def_version, const FunctionLibraryDefinition* lib_def,
951 device_mgr, env, device, graph_def_version, lib_def, optimizer_options,
957 int graph_def_version, const FunctionLibraryDefinition* lib_def,
960 return NewFunctionLibraryRuntime(device_mgr, env, device, graph_def_version,
245 FunctionLibraryRuntimeImpl( const DeviceMgr* dmgr, Env* env, Device* device, int graph_def_version, const FunctionLibraryDefinition* lib_def, const OptimizerOptions& optimizer_options, CustomKernelCreator custom_kernel_creator, ProcessFunctionLibraryRuntime* parent) argument
944 NewFunctionLibraryRuntime( const DeviceMgr* device_mgr, Env* env, Device* device, int graph_def_version, const FunctionLibraryDefinition* lib_def, const OptimizerOptions& optimizer_options, CustomKernelCreator custom_kernel_creator, ProcessFunctionLibraryRuntime* parent) argument
955 NewFunctionLibraryRuntime( const DeviceMgr* device_mgr, Env* env, Device* device, int graph_def_version, const FunctionLibraryDefinition* lib_def, const OptimizerOptions& optimizer_options, ProcessFunctionLibraryRuntime* parent) argument
/external/tensorflow/tensorflow/core/framework/
H A Dop_def_util.h34 Status CheckOpDeprecation(const OpDef& op_def, int graph_def_version);
H A Dshape_inference_testutil.h40 int graph_def_version = TF_GRAPH_DEF_VERSION; member in struct:tensorflow::ShapeInferenceTestOp
H A 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,
610 int graph_def_version() const { return graph_def_version_; } function in class:tensorflow::shape_inference::InferenceContext
H A Dop_kernel.cc98 graph_def_version_(context->graph_def_version()),
106 context->graph_def_version()));
188 const MemoryTypeSlice& output_memory_types, int graph_def_version,
200 graph_def_version_(graph_def_version),
1101 const NodeDef& node_def, int graph_def_version, Status* status) {
1104 node_def, graph_def_version, &kernel);
1110 const NodeDef& node_def, int graph_def_version,
1167 input_memory_types, outputs, output_memory_types, graph_def_version, &s);
183 OpKernelConstruction( DeviceType device_type, DeviceBase* device, Allocator* allocator, const NodeDef* node_def, const OpDef* op_def, FunctionLibraryRuntime* flib, const DataTypeSlice& input_types, const MemoryTypeSlice& input_memory_types, const DataTypeSlice& output_types, const MemoryTypeSlice& output_memory_types, int graph_def_version, Status* status) argument
1099 CreateOpKernel( DeviceType device_type, DeviceBase* device, Allocator* allocator, const NodeDef& node_def, int graph_def_version, Status* status) argument
1108 CreateOpKernel(DeviceType device_type, DeviceBase* device, Allocator* allocator, FunctionLibraryRuntime* flib, const NodeDef& node_def, int graph_def_version, OpKernel** kernel) argument
H A Dshape_inference.cc33 int graph_def_version, const NodeDef* node_def, const OpDef& op_def,
40 : graph_def_version_(graph_def_version),
86 int graph_def_version, const NodeDef* node_def, const OpDef& op_def,
93 : graph_def_version_(graph_def_version),
137 int graph_def_version, const NodeDef* node_def, const OpDef& op_def,
143 : graph_def_version_(graph_def_version),
32 InferenceContext( int graph_def_version, const NodeDef* node_def, const OpDef& op_def, const std::vector<TensorShapeProto>& input_shapes, const std::vector<const Tensor*>& input_tensors, const std::vector<TensorShapeProto>& input_tensors_as_shapes, const std::vector< std::unique_ptr<std::vector<std::pair<TensorShapeProto, DataType>>>>& input_handle_shapes_and_types) argument
85 InferenceContext( int graph_def_version, const NodeDef* node_def, const OpDef& op_def, const std::vector<PartialTensorShape>& input_shapes, const std::vector<const Tensor*>& input_tensors, const std::vector<PartialTensorShape>& input_tensors_as_shapes, const std::vector< std::unique_ptr<std::vector<std::pair<PartialTensorShape, DataType>>>>& input_handle_shapes_and_types) argument
136 InferenceContext( int graph_def_version, const NodeDef* node_def, const OpDef& op_def, const std::vector<ShapeHandle>& input_shapes, const std::vector<const Tensor*>& input_tensors, const std::vector<ShapeHandle>& input_tensors_as_shapes, std::vector<std::unique_ptr<std::vector<ShapeAndType>>> input_handle_shapes_and_types) argument
H A Dshape_inference_testutil.cc64 op.graph_def_version, &op.node_def, op_reg_data->op_def, in_shapes,
H A Dop_kernel.h241 int graph_def_version, Status* status);
324 int graph_def_version() const { return graph_def_version_; } function in class:tensorflow::OpKernelConstruction
1164 int graph_def_version, Status* status);
1167 const NodeDef& def, int graph_def_version,
/external/tensorflow/tensorflow/compiler/jit/
H A Dcreate_xla_launch_op.cc95 fbody->ret_types, output_memory_types, flr->graph_def_version(), &s);
/external/tensorflow/tensorflow/core/kernels/
H A Dops_testutil.h102 Status InitOpWithGraphVersion(int graph_def_version) { argument
105 node_def_, graph_def_version, &status);
/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compiler.h239 int graph_def_version = TF_GRAPH_DEF_VERSION; member in struct:tensorflow::XlaCompiler::Options
/external/tensorflow/tensorflow/compiler/jit/kernels/
H A Dxla_launch_op.cc259 options.graph_def_version = ctx->function_library()->graph_def_version();
/external/tensorflow/tensorflow/core/kernels/data/
H A Dfilter_dataset_op.cc33 graph_def_version_(ctx->graph_def_version()) {
H A Dmap_dataset_op.cc33 graph_def_version_(ctx->graph_def_version()) {
H A Dflat_map_dataset_op.cc34 graph_def_version_(ctx->graph_def_version()) {

Completed in 1630 milliseconds

12