Searched refs:function_def (Results 1 – 9 of 9) sorted by relevance
| /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 | 173 const FunctionDef* function_def = lib_def->Find(function_name); in EnsureFunctionIsStateless() local
174 if (!function_def) { in EnsureFunctionIsStateless()
178 for (const NodeDef& node_def : function_def->node_def()) { in EnsureFunctionIsStateless()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | meta_graph.py | 478 for function_def in meta_graph_def.graph_def.library.function:
479 op_name_to_function[function_def.signature.name] = function_def
503 for function_def in meta_graph_def.graph_def.library.function:
504 for function_node_def in function_def.node_def:
|
| D | meta_graph_test.py | 231 for function_def in meta_graph_def.graph_def.library.function:
232 for node_def in function_def.node_def:
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 132 const tensorflow::FunctionDef* function_def, bool keep_nested_shapes, in InferShapesForFunction() argument
135 auto it = functions_.find(function_def); in InferShapesForFunction()
141 *function_def, outer_context->get_context()->attrs(), in InferShapesForFunction()
152 functions_[function_def].reset(new_graph); in InferShapesForFunction()
181 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/eager/ |
| D | function.py | 278 function_def = function_pb2.FunctionDef()
279 function_def.ParseFromString(compat.as_bytes(proto_data))
282 self.definition = function_def
283 self.name = function_def.signature.name
284 self.signature = function_def.signature
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 798 string function_def = MatMulFunction(); in TEST() local
799 TFE_ContextAddFunctionDef(ctx, function_def.data(), function_def.size(), in TEST()
839 string function_def = MatMulFunction(); in BM_ExecuteFunction() local
840 TFE_ContextAddFunctionDef(ctx, function_def.data(), function_def.size(), in BM_ExecuteFunction()
|
| D | c_api.cc | 893 tensorflow::FunctionDef function_def; in TFE_ContextAddFunctionDef() local
894 if (!function_def.ParseFromArray(serialized_function_def, size)) { in TFE_ContextAddFunctionDef()
900 status->status = ctx->func_lib_def.AddFunctionDef(function_def); in TFE_ContextAddFunctionDef()
|