/external/tensorflow/tensorflow/python/framework/ |
H A D | cpp_shape_inference.h | 45 int graph_def_version, const string& serialized_node_def,
|
H A D | cpp_shape_inference.cc | 52 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 D | test_ops.cc | 48 if (c->graph_def_version() != TF_GRAPH_DEF_VERSION - 1) { 133 : OpKernel(ctx), graph_def_version_(ctx->graph_def_version()) {}
|
H A D | common_shapes.py | 653 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 D | function.h | 57 int graph_def_version, const FunctionLibraryDefinition* lib_def, 67 int graph_def_version, const FunctionLibraryDefinition* lib_def,
|
H A D | process_function_library_runtime.h | 33 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 D | process_function_library_runtime.cc | 29 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 D | kernel_benchmark_testlib.cc | 73 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 D | shape_refiner.h | 91 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 D | executor.h | 232 const NodeDef& ndef, int graph_def_version,
|
H A D | function.cc | 145 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 D | op_def_util.h | 34 Status CheckOpDeprecation(const OpDef& op_def, int graph_def_version);
|
H A D | shape_inference_testutil.h | 40 int graph_def_version = TF_GRAPH_DEF_VERSION; member in struct:tensorflow::ShapeInferenceTestOp
|
H A D | shape_inference.h | 173 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 D | op_kernel.cc | 98 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 D | shape_inference.cc | 33 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 D | shape_inference_testutil.cc | 64 op.graph_def_version, &op.node_def, op_reg_data->op_def, in_shapes,
|
H A D | op_kernel.h | 241 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 D | create_xla_launch_op.cc | 95 fbody->ret_types, output_memory_types, flr->graph_def_version(), &s);
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | ops_testutil.h | 102 Status InitOpWithGraphVersion(int graph_def_version) { argument 105 node_def_, graph_def_version, &status);
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | xla_compiler.h | 239 int graph_def_version = TF_GRAPH_DEF_VERSION; member in struct:tensorflow::XlaCompiler::Options
|
/external/tensorflow/tensorflow/compiler/jit/kernels/ |
H A D | xla_launch_op.cc | 259 options.graph_def_version = ctx->function_library()->graph_def_version();
|
/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | filter_dataset_op.cc | 33 graph_def_version_(ctx->graph_def_version()) {
|
H A D | map_dataset_op.cc | 33 graph_def_version_(ctx->graph_def_version()) {
|
H A D | flat_map_dataset_op.cc | 34 graph_def_version_(ctx->graph_def_version()) {
|