| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | graph_rewrite_dataset.cc | 101 void AddFakeSinks(FunctionDef* function_def) { in AddFakeSinks() argument
103 for (const auto& output : function_def->signature().output_arg()) { in AddFakeSinks()
104 NodeDef* node = function_def->add_node_def(); in AddFakeSinks()
106 strings::StrCat("FakeSink", counter++), function_def, node); in AddFakeSinks()
108 node->add_input(function_def->ret().at(output.name())); in AddFakeSinks()
111 (*function_def->mutable_ret())[output.name()] = in AddFakeSinks()
116 void RemoveFakeSinks(FunctionDef* function_def) { in RemoveFakeSinks() argument
119 for (const auto& node : function_def->node_def()) { in RemoveFakeSinks()
124 for (const auto& output_arg : function_def->signature().output_arg()) { in RemoveFakeSinks()
125 const string& tensor = function_def->ret().at(output_arg.name()); in RemoveFakeSinks()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_api_info.cc | 29 Status FunctionApiInfo::Init(const FunctionDef& function_def) { in Init() argument
31 for (const auto& attr : function_def.attr()) { in Init()
48 input_arg_dtypes_.reserve(function_def.signature().input_arg_size()); in Init()
49 for (const auto& input_arg : function_def.signature().input_arg()) { in Init()
52 output_arg_dtypes_.reserve(function_def.signature().output_arg_size()); in Init()
53 for (const auto& output_arg : function_def.signature().output_arg()) { in Init()
59 "Function '", function_def.signature().name(), in Init()
|
| D | function_api_info.h | 42 Status Init(const FunctionDef& function_def);
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 83 const FunctionDef* function_def; in Init() local
88 function_def = pflr_->GetFLR(host_cpu_device_->name()) in Init()
92 function_def = flr_->GetFunctionLibraryDefinition()->Find(ndef.op()); in Init()
95 if (function_def != nullptr) { in Init()
96 op_def = &(function_def->signature()); in Init()
139 options.config_proto, function_def->signature().name(), in Init()
|
| D | execute.cc | 229 const FunctionDef* function_def = in GetOutputDTypes() local
231 if (function_def != nullptr) { in GetOutputDTypes()
232 op_def = &(function_def->signature()); in GetOutputDTypes()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils.cc | 175 const FunctionDef& function_def, bool skip_assert) { in IsFunctionStateful() argument
176 if (!function_def.signature().is_stateful()) return false; in IsFunctionStateful()
178 for (const NodeDef& node_def : function_def.node_def()) { in IsFunctionStateful()
|
| D | function_utils.h | 110 const FunctionDef& function_def,
|
| D | vectorization_utils_test.cc | 118 string GetRetval(const FunctionDef& function_def, int index) { in GetRetval() argument
119 return function_def.ret().at( in GetRetval()
120 function_def.signature().output_arg(index).name()); in GetRetval()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_def_util_test.cc | 246 FunctionDef* function_def = graph_def.mutable_library()->add_function(); in TEST() local
247 function_def->mutable_signature()->set_name("F"); in TEST()
249 function_def->add_node_def()->set_op(op); in TEST()
|
| D | dataset.h | 202 const FunctionDef* function_def = flib_def.Find(function_name); in EnsureFunctionIsStateless() local
203 if (!function_def) { in EnsureFunctionIsStateless()
207 for (const NodeDef& node_def : function_def->node_def()) { in EnsureFunctionIsStateless()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | op_hint.py | 784 def _find_children_hints_in_while_loop(function_def, nodes_mapping): argument
797 for node in function_def.node_def:
802 name_to_seq_num = _extract_topology_sequence_mapping(function_def.node_def)
853 for function_def in graph_def.library.function:
854 if function_def.signature.name == body_name:
855 function_inputs = function_def.signature.input_arg
863 function_def, nodes_mapping)
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op.cc | 70 const FunctionDef* function_def = in CompilationRequested() local
72 if (function_def == nullptr) { in CompilationRequested()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 133 const tensorflow::FunctionDef* function_def, bool keep_nested_shapes, in InferShapesForFunction() argument
136 auto it = functions_.find(function_def); in InferShapesForFunction()
142 *function_def, outer_context->get_context()->attrs(), in InferShapesForFunction()
153 functions_[function_def].reset(new_graph); in InferShapesForFunction()
182 for (const auto& node_def : function_def->node_def()) { in InferShapesForFunction()
|
| D | shape_refiner.h | 214 Status InferShapesForFunction(const tensorflow::FunctionDef* function_def,
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | meta_graph.py | 481 for function_def in meta_graph_def.graph_def.library.function:
482 op_name_to_function[function_def.signature.name] = function_def
506 for function_def in meta_graph_def.graph_def.library.function:
507 for function_node_def in function_def.node_def:
|
| D | function_def_to_graph_test.py | 242 fdef = fn.get_concrete_function(inp).function_def
|
| D | meta_graph_test.py | 235 for function_def in meta_graph_def.graph_def.library.function:
236 for node_def in function_def.node_def:
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 343 function_def = function_pb2.FunctionDef()
344 function_def.ParseFromString(compat.as_bytes(proto_data))
348 self.definition = function_def
349 self.name = compat.as_bytes(function_def.signature.name)
350 self.signature = function_def.signature
698 def function_def(self): member in ConcreteFunction
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 841 tensorflow::FunctionDef function_def; in TFE_ContextAddFunctionDef() local
842 if (!function_def.ParseFromArray(serialized_function_def, size)) { in TFE_ContextAddFunctionDef()
847 status->status = ctx->context.AddFunctionDef(function_def); in TFE_ContextAddFunctionDef()
|
| D | c_api_test.cc | 1346 string function_def = MatMulFunction(); in FunctionDefAndExecute() local
1347 TFE_ContextAddFunctionDef(ctx, function_def.data(), function_def.size(), in FunctionDefAndExecute()
1392 string function_def = MatMulFunction(); in BM_ExecuteFunction() local
1393 TFE_ContextAddFunctionDef(ctx, function_def.data(), function_def.size(), in BM_ExecuteFunction()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 272 return context->Context()->AddFunctionDef(request->function_def()); in RegisterFunction()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | eager_service.proto | 126 FunctionDef function_def = 2; field
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1041 const FunctionDef* function_def = in AddFunction() local
1046 MakeGrapplerFunctionItem(*function_def, function_library_, in AddFunction()
1064 fun_to_grappler_function_item_[function_def->signature().name()] = in AddFunction()
|