Home
last modified time | relevance | path

Searched refs:op_reg_data (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dop.cc40 const OpRegistrationData* op_reg_data = nullptr; in LookUpOpDef() local
41 TF_RETURN_IF_ERROR(LookUp(op_type_name, &op_reg_data)); in LookUpOpDef()
42 *op_def = &op_reg_data->op_def; in LookUpOpDef()
62 const OpRegistrationData** op_reg_data) const { in LookUp()
68 *op_reg_data = res; in LookUp()
73 return LookUpSlow(op_type_name, op_reg_data); in LookUp()
77 const OpRegistrationData** op_reg_data) const { in LookUpSlow()
78 *op_reg_data = nullptr; in LookUpSlow()
121 *op_reg_data = res; in LookUpSlow()
221 std::unique_ptr<OpRegistrationData> op_reg_data(new OpRegistrationData); in RegisterAlreadyLocked() local
[all …]
Dop_def_builder_test.cc45 OpRegistrationData op_reg_data; in ExpectSuccess() local
46 Status status = builder.Finalize(&op_reg_data); in ExpectSuccess()
48 OpDef& op_def = op_reg_data.op_def; in ExpectSuccess()
59 *shape_fn_out = op_reg_data.shape_inference_fn; in ExpectSuccess()
65 OpRegistrationData op_reg_data; in ExpectOrdered() local
66 Status status = builder.Finalize(&op_reg_data); in ExpectOrdered()
68 OpDef& op_def = op_reg_data.op_def; in ExpectOrdered()
78 OpRegistrationData op_reg_data; in ExpectFailure() local
79 Status status = builder.Finalize(&op_reg_data); in ExpectFailure()
630 OpRegistrationData op_reg_data; in TEST_F() local
[all …]
Dcommon_shape_fns_test.cc38 OpRegistrationData op_reg_data; in MakeOpDef() local
46 CHECK(b.Attr("foo:string").Finalize(&op_reg_data).ok()); in MakeOpDef()
47 return op_reg_data.op_def; in MakeOpDef()
53 OpRegistrationData op_reg_data; in TEST() local
57 .Finalize(&op_reg_data)); in TEST()
58 OpDef op_def = op_reg_data.op_def; in TEST()
73 OpRegistrationData op_reg_data; in TEST() local
77 .Finalize(&op_reg_data)); in TEST()
78 OpDef op_def = op_reg_data.op_def; in TEST()
101 OpRegistrationData op_reg_data; in TEST() local
[all …]
Dshape_inference_testutil.cc32 const OpRegistrationData* op_reg_data; in InferShapes() local
33 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUp(op.name, &op_reg_data)); in InferShapes()
64 op.graph_def_version, &op.node_def, op_reg_data->op_def, in_shapes, in InferShapes()
67 if (op_reg_data->shape_inference_fn == nullptr) { in InferShapes()
73 TF_RETURN_IF_ERROR(c.Run(op_reg_data->shape_inference_fn)); in InferShapes()
Dop.h49 const OpRegistrationData** op_reg_data) const = 0;
75 const OpRegistrationData** op_reg_data) const override;
148 const OpRegistrationData** op_reg_data) const;
173 const OpRegistrationData** op_reg_data) const override;
Dop_registration_test.cc28 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register() argument
29 op_reg_data->op_def.set_name(op_name); in Register()
Dgraph_def_util_test.cc32 OpRegistrationData op_reg_data; in FinalizeOpDef() local
33 const Status s = b.Finalize(&op_reg_data); in FinalizeOpDef()
34 *op_def = op_reg_data.op_def; in FinalizeOpDef()
Ddataset.cc277 const OpRegistrationData* op_reg_data = nullptr; in AddFunction() local
278 TF_RETURN_IF_ERROR(ctx->flib_def().LookUp(node_def.op(), &op_reg_data)); in AddFunction()
279 if (op_reg_data->is_function_op) { in AddFunction()
280 TF_RETURN_IF_ERROR(AddFunction(ctx, op_reg_data->op_def.name())); in AddFunction()
Dop_def_util_test.cc46 OpRegistrationData op_reg_data; in TestBuilder() local
47 Status status = builder.Finalize(&op_reg_data); in TestBuilder()
52 return ValidateOpDef(op_reg_data.op_def); in TestBuilder()
Dop_def_builder.h152 Status Finalize(OpRegistrationData* op_reg_data) const;
Dnode_def_util_test.cc35 OpRegistrationData op_reg_data; in ToOpDef() local
36 TF_EXPECT_OK(builder.Finalize(&op_reg_data)); in ToOpDef()
37 return op_reg_data.op_def; in ToOpDef()
Dfunction.cc1267 const string& op, const OpRegistrationData** op_reg_data) const { in LookUp()
1271 *op_reg_data = &iter->second->op_registration_data; in LookUp()
1274 return default_registry_->LookUp(op, op_reg_data); in LookUp()
1551 OpRegistrationData op_reg_data; in Create() local
1552 TF_CHECK_OK(b.Finalize(&op_reg_data)); in Create()
1553 fdef.mutable_signature()->Swap(&op_reg_data.op_def); in Create()
1607 OpRegistrationData op_reg_data; in Define() local
1608 TF_CHECK_OK(b.Finalize(&op_reg_data)); in Define()
1609 fdef.mutable_signature()->Swap(&op_reg_data.op_def); in Define()
/external/tensorflow/tensorflow/core/ops/
Dtpu_embedding_ops.cc78 OpRegistrationData* op_reg_data) { in RegisterPerTableLoadOpsForAlgorithmBody() argument
88 auto* op_def = &op_reg_data->op_def; in RegisterPerTableLoadOpsForAlgorithmBody()
188 op_reg_data->shape_inference_fn = shape_inference_function; in RegisterPerTableLoadOpsForAlgorithmBody()
194 OpRegistrationData* op_reg_data) { in RegisterPerTableRetrieveOpsForAlgorithmBody() argument
205 auto* op_def = &op_reg_data->op_def; in RegisterPerTableRetrieveOpsForAlgorithmBody()
284 op_reg_data->shape_inference_fn = shape_inference_function; in RegisterPerTableRetrieveOpsForAlgorithmBody()
292 [alg](OpRegistrationData* op_reg_data) -> Status { in RegisterPerTableLoadAndRetrieveOps() argument
294 op_reg_data); in RegisterPerTableLoadAndRetrieveOps()
300 [alg](OpRegistrationData* op_reg_data) -> Status { in RegisterPerTableLoadAndRetrieveOps() argument
302 op_reg_data); in RegisterPerTableLoadAndRetrieveOps()
[all …]
Dshape_function_test.cc28 for (const OpRegistrationData& op_reg_data : op_data) { in TEST() local
29 EXPECT_TRUE(op_reg_data.shape_inference_fn != nullptr) in TEST()
30 << op_reg_data.op_def.name(); in TEST()
Dmath_ops_test.cc203 const OpRegistrationData* op_reg_data; in TEST() local
204 TF_ASSERT_OK(OpRegistry::Global()->LookUp(op.name, &op_reg_data)); in TEST()
209 CHECK(op_reg_data->shape_inference_fn != nullptr); in TEST()
211 TF_GRAPH_DEF_VERSION, &op.node_def, op_reg_data->op_def, in TEST()
215 Status s = c->Run(op_reg_data->shape_inference_fn); in TEST()
/external/tensorflow/tensorflow/core/graph/
Dvalidate_test.cc99 OpRegistrationData op_reg_data; in TEST() local
100 TF_ASSERT_OK(OpDefBuilder("UniqueSnowflake").Finalize(&op_reg_data)); in TEST()
102 *op_list.add_op() = op_reg_data.op_def; in TEST()
111 OpRegistrationData op_reg_data; in TEST() local
112 TF_ASSERT_OK(OpDefBuilder("NotAnywhere").Finalize(&op_reg_data)); in TEST()
114 *op_list.add_op() = op_reg_data.op_def; in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc245 const OpRegistrationData* op_reg_data; in AddNode() local
246 TF_RETURN_IF_ERROR(ops_registry_->LookUp(node->type_string(), &op_reg_data)); in AddNode()
247 if (op_reg_data->shape_inference_fn == nullptr && in AddNode()
271 TF_RETURN_IF_ERROR(RunShapeFn(node, op_reg_data, ec.get())); in AddNode()
399 const OpRegistrationData* op_reg_data; in UpdateNode() local
400 TF_RETURN_IF_ERROR(ops_registry_->LookUp(node->type_string(), &op_reg_data)); in UpdateNode()
401 if (op_reg_data->shape_inference_fn == nullptr && in UpdateNode()
408 if (!op_reg_data->shape_inference_fn) { in UpdateNode()
413 return RunShapeFn(node, op_reg_data, node_ext_context); in UpdateNode()
622 const OpRegistrationData* op_reg_data, in RunShapeFn() argument
[all …]
/external/tensorflow/tensorflow/python/util/
Dkernel_registry.cc35 const tensorflow::OpRegistrationData* op_reg_data; in TryFindKernelClass() local
37 tensorflow::OpRegistry::Global()->LookUp(node_def.op(), &op_reg_data); in TryFindKernelClass()
42 AddDefaultsToNodeDef(op_reg_data->op_def, &node_def); in TryFindKernelClass()
/external/tensorflow/tensorflow/python/framework/
Dcpp_shape_inference.cc65 const OpRegistrationData* op_reg_data; in RunCppShapeInferenceImpl() local
66 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUp(node.op(), &op_reg_data)); in RunCppShapeInferenceImpl()
68 if (op_reg_data->shape_inference_fn == nullptr) { in RunCppShapeInferenceImpl()
128 graph_def_version, &node, op_reg_data->op_def, input_shapes, in RunCppShapeInferenceImpl()
133 TF_RETURN_IF_ERROR(c.Run(op_reg_data->shape_inference_fn)); in RunCppShapeInferenceImpl()
/external/tensorflow/tensorflow/python/grappler/
Dmodel_analyzer.cc82 const OpRegistrationData* op_reg_data; in PrintNodeInfo() local
83 Status status = OpRegistry::Global()->LookUp(node->op(), &op_reg_data); in PrintNodeInfo()
86 } else if (!op_reg_data->shape_inference_fn) { in PrintNodeInfo()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc49 const tensorflow::OpRegistrationData* op_reg_data = nullptr; in GetNumRetvals() local
50 auto status = tensorflow::OpRegistry::Global()->LookUp(op_name, &op_reg_data); in GetNumRetvals()
52 status = context->FindFunctionOpData(op_name, &op_reg_data); in GetNumRetvals()
56 const tensorflow::OpDef& op_def = op_reg_data->op_def; in GetNumRetvals()
/external/tensorflow/tensorflow/core/api_def/
Dupdate_api_def.cc141 OpRegistrationData op_reg_data; in ValidateOpDocs() local
142 TF_CHECK_OK(b.Finalize(&op_reg_data)); in ValidateOpDocs()
143 return CheckDocsMatch(op, op_reg_data.op_def); in ValidateOpDocs()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.cc57 const OpRegistrationData* op_reg_data = nullptr; in OpDefForOp() local
58 Status s = OpRegistry::Global()->LookUp(op_name, &op_reg_data); in OpDefForOp()
60 *op_def = &op_reg_data->op_def; in OpDefForOp()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc225 const tensorflow::OpRegistrationData* op_reg_data; in InitializeNodeDef() local
227 tensorflow::OpRegistry::Global()->LookUp(nodedef_.op(), &op_reg_data)); in InitializeNodeDef()
228 AddDefaultsToNodeDef(op_reg_data->op_def, &nodedef_); in InitializeNodeDef()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc83 const OpRegistrationData* op_reg_data = nullptr; in IsOutputPortRefValue() local
84 Status s = op_registry.LookUp(node.op(), &op_reg_data); in IsOutputPortRefValue()
87 s = OutputTypeForNode(node, op_reg_data->op_def, port_id, &output_type); in IsOutputPortRefValue()

12