| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_dataset_op_test.cc | 172 FunctionDef func_def = test::function::XTimesTwo(); in TEST_F() local
175 TF_ASSERT_OK(InitFunctionLibraryRuntime({func_def}, cpu_num)); in TEST_F()
184 range_dataset->node_name(), func_def.signature().name(), &map_kernel)); in TEST_F()
199 FunctionDef func_def = test::function::XTimesTwo(); in TEST_F() local
202 TF_ASSERT_OK(InitFunctionLibraryRuntime({func_def}, cpu_num)); in TEST_F()
211 range_dataset->node_name(), func_def.signature().name(), &map_kernel)); in TEST_F()
227 FunctionDef func_def = test::function::XTimesTwo(); in TEST_F() local
230 TF_ASSERT_OK(InitFunctionLibraryRuntime({func_def}, cpu_num)); in TEST_F()
239 range_dataset->node_name(), func_def.signature().name(), &map_kernel)); in TEST_F()
278 FunctionDef func_def = test::function::XTimesTwo(); in TEST_P() local
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_def_util.cc | 130 for (const auto& func_def : graph_def.library().function()) { in IsFunction() local
131 if (op_name == func_def.signature().name()) return true; in IsFunction()
151 FunctionDef* func_def = graph_def->mutable_library()->mutable_function(f); in RemoveNewDefaultAttrsFromGraphDef() local
152 for (int n = 0; n < func_def->node_def_size(); ++n) { in RemoveNewDefaultAttrsFromGraphDef()
153 NodeDef* node_def = func_def->mutable_node_def(n); in RemoveNewDefaultAttrsFromGraphDef()
|
| D | function.cc | 766 string DebugString(const FunctionDef& func_def) { return Print(func_def); } in DebugString() argument
1443 const auto& func_def = flib.Find(func_name); in ReachableFunctions() local
1444 const auto attr_it = func_def->attr().find(kApiImplements); in ReachableFunctions()
1445 if (attr_it != func_def->attr().end()) { in ReachableFunctions()
|
| D | function_test.cc | 1357 auto func_def = FDH::Create( in TEST() local
1364 auto* attr = func_def.mutable_attr(); in TEST()
1367 return func_def; in TEST()
|
| D | function.h | 241 string DebugString(const FunctionDef& func_def);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_api_info_test.cc | 53 FunctionDef* func_def) { in PopulateFunction() argument
54 OpDef* sig = func_def->mutable_signature(); in PopulateFunction()
59 auto* func_attr = func_def->mutable_attr(); in PopulateFunction()
|
| D | meta_optimizer.cc | 630 for (const FunctionDef& func_def : in Optimize() local
632 if (flib.Find(func_def.signature().name()) == nullptr) { in Optimize()
633 TF_RETURN_IF_ERROR(flib.AddFunctionDef(func_def)); in Optimize()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | rebatch.cc | 215 for (const FunctionDef& func_def : in RecursivelyHandleOp() local
217 if (flib->Find(func_def.signature().name()) == nullptr) { in RecursivelyHandleOp()
218 TF_RETURN_IF_ERROR(flib->AddFunctionDef(func_def)); in RecursivelyHandleOp()
|
| /external/sl4a/Docs/ |
| D | generate_api_reference_md.py | 49 def __init__(self, rpc_def, func_def): argument
57 self._ParseFunctionDefinition(func_def)
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass.cc | 281 const FunctionDef* func_def = flib_def.Find(function_name); in InferDeviceForCluster() local
282 TF_RET_CHECK(func_def) << "Could not find " << function_name; in InferDeviceForCluster()
285 for (const NodeDef& ndef : func_def->node_def()) { in InferDeviceForCluster()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 642 auto* func_def = function_library_->Find(op_reg_data->op_def.name()); in RunShapeFn() local
643 if (func_def) { in RunShapeFn()
644 return InferShapesForFunction(func_def, keep_nested_shape_inferences_, in RunShapeFn()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.h | 120 bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def);
|
| D | c_test_util.cc | 412 bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def) { in GetFunctionDef() argument
418 if (ret) ret = func_def->ParseFromArray(buffer->data, buffer->length); in GetFunctionDef()
|
| D | c_api_experimental.cc | 186 const char* func_def = R"PREFIX( in CreateFakeDatasetFunction() local
253 func_def, /*mutate_proto_func*/ nullptr, status); in CreateFakeDatasetFunction()
271 const char* func_def = R"PREFIX( in CreateImagenetDatasetFunctions()
7148 return CreateFunctionsFromTextProto(func_def, &mutate_proto_func, status); in CreateImagenetDatasetFunctions()
7167 const char* func_def = R"PREFIX( in CreateMNISTDatasetFunctions()
8295 return CreateFunctionsFromTextProto(func_def, &mutate_proto_func, status); in CreateMNISTDatasetFunctions()
|
| /external/python/mock/mock/tests/ |
| D | testhelpers.py | 408 func_def = "def foo(a, *, b=None):\n pass\n"
410 exec (func_def, namespace)
|