| /external/tensorflow/tensorflow/core/framework/ |
| D | op.h | 48 virtual Status LookUp(const string& op_type_name,
74 Status LookUp(const string& op_type_name,
172 Status LookUp(const string& op_type_name,
|
| D | op.cc | 41 TF_RETURN_IF_ERROR(LookUp(op_type_name, &op_reg_data)); in LookUpOpDef()
61 Status OpRegistry::LookUp(const string& op_type_name, in LookUp() function in tensorflow::OpRegistry
263 Status OpListOpRegistry::LookUp(const string& op_type_name, in LookUp() function in tensorflow::OpListOpRegistry
|
| D | shape_inference_testutil.cc | 33 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUp(op.name, &op_reg_data)); in InferShapes()
|
| D | dataset.cc | 278 TF_RETURN_IF_ERROR(ctx->flib_def().LookUp(node_def.op(), &op_reg_data)); in AddFunction()
|
| D | function_test.cc | 989 TEST(FunctionLibraryDefinitionTest, LookUp) { in TEST() argument
1003 TF_EXPECT_OK(lib_def.LookUp("XTimesTwo", &op_reg_data)); in TEST()
|
| D | function.h | 389 Status LookUp(const string& op_type_name,
|
| D | op_kernel.cc | 1180 const Status s = OpRegistry::Global()->LookUp(def.op(), &op_reg_data); in SupportedDeviceTypesForNode()
1352 const Status status = op_registry.LookUp(kernel_def.op(), &op_reg_data); in ValidateKernelRegistrations()
|
| D | function.cc | 1266 Status FunctionLibraryDefinition::LookUp( in LookUp() function in tensorflow::FunctionLibraryDefinition
1274 return default_registry_->LookUp(op, op_reg_data); in LookUp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compilation_cache.h | 45 StatusOr<std::shared_ptr<Executable>> LookUp(
|
| D | compilation_cache.cc | 54 StatusOr<std::shared_ptr<Executable>> CompilationCache::LookUp( in LookUp() function in xla::CompilationCache
|
| D | service.cc | 861 compilation_cache_.LookUp(arg->handle())); in Execute()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | kernel_registry.cc | 37 tensorflow::OpRegistry::Global()->LookUp(node_def.op(), &op_reg_data); in TryFindKernelClass()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | model_analyzer.cc | 83 Status status = OpRegistry::Global()->LookUp(node->op(), &op_reg_data); in PrintNodeInfo()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.cc | 66 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUp(node.op(), &op_reg_data)); in RunCppShapeInferenceImpl()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.cc | 58 Status s = OpRegistry::Global()->LookUp(op_name, &op_reg_data); in OpDefForOp()
|
| D | context.cc | 238 return func_lib_def_.LookUp(name, op_data); in FindFunctionOpData()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 246 TF_RETURN_IF_ERROR(ops_registry_->LookUp(node->type_string(), &op_reg_data)); in AddNode()
400 TF_RETURN_IF_ERROR(ops_registry_->LookUp(node->type_string(), &op_reg_data)); in UpdateNode()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 50 auto status = tensorflow::OpRegistry::Global()->LookUp(op_name, &op_reg_data); in GetNumRetvals()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 54 TF_RETURN_IF_ERROR(flib.LookUp(node.op(), ®istration)); in RegisterFunctionBodyOutputs()
585 TF_RETURN_IF_ERROR(flib.LookUp(func_def_node.op(), ®istration)); in MakeGrapplerFunctionItem()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 227 tensorflow::OpRegistry::Global()->LookUp(nodedef_.op(), &op_reg_data)); in InitializeNodeDef()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops_test.cc | 204 TF_ASSERT_OK(OpRegistry::Global()->LookUp(op.name, &op_reg_data)); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner.cc | 84 Status s = op_registry.LookUp(node.op(), &op_reg_data); in IsOutputPortRefValue()
|
| D | loop_optimizer.cc | 208 TF_RETURN_IF_ERROR(op_registry->LookUp(node->op(), &op_reg_data)); in HandleInvariantNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization.cc | 155 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUp(node.op(), &op_reg_data)); in GetInputMap()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 1882 TF_RETURN_IF_ERROR(library->LookUp(node_in->type_string(), &op_reg_data)); in CopyShapeInferenceNodeToGraph()
2310 TF_RETURN_IF_ERROR(library->LookUp(node->type_string(), &op_reg_data)); in MakePrunedGraphCopyAndInline()
|