Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/framework/
Dop.cc46 const OpRegistrationData* op_reg_data = nullptr; in LookUpOpDef() local
47 TF_RETURN_IF_ERROR(LookUp(op_type_name, &op_reg_data)); in LookUpOpDef()
48 *op_def = &op_reg_data->op_def; in LookUpOpDef()
85 const OpRegistrationData** op_reg_data) const { in LookUp()
86 if ((*op_reg_data = LookUp(op_type_name))) return Status::OK(); in LookUp()
235 std::unique_ptr<OpRegistrationData> op_reg_data(new OpRegistrationData); in RegisterAlreadyLocked() local
236 Status s = op_data_factory(op_reg_data.get()); in RegisterAlreadyLocked()
238 s = ValidateOpDef(op_reg_data->op_def); in RegisterAlreadyLocked()
240 !gtl::InsertIfNotPresent(&registry_, op_reg_data->op_def.name(), in RegisterAlreadyLocked()
241 op_reg_data.get())) { in RegisterAlreadyLocked()
[all …]
Dnode_properties_test.cc26 OpRegistrationData op_reg_data; in ToOpDef() local
27 EXPECT_TRUE(builder.Finalize(&op_reg_data).ok()); in ToOpDef()
28 return op_reg_data.op_def; in ToOpDef()
44 const OpRegistrationData** op_reg_data) const override { in LookUp()
46 *op_reg_data = &op_reg_; in LookUp()
49 *op_reg_data = nullptr; in LookUp()
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()
63 op.graph_def_version, op.node_def, op_reg_data->op_def, in_shapes, in InferShapes()
66 if (op_reg_data->shape_inference_fn == nullptr) { in InferShapes()
72 TF_RETURN_IF_ERROR(c.Run(op_reg_data->shape_inference_fn)); in InferShapes()
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()
Dop.h49 const OpRegistrationData** op_reg_data) const = 0;
76 const OpRegistrationData** op_reg_data) const override;
184 const OpRegistrationData** op_reg_data) const override;
Dgraph_def_util.cc173 const OpRegistrationData* op_reg_data = nullptr; in StripDefaultAttributes() local
174 Status s = op_registry.LookUp(node->op(), &op_reg_data); in StripDefaultAttributes()
182 op_def = &op_reg_data->op_def; in StripDefaultAttributes()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dshape_inference.cc32 const tensorflow::OpRegistrationData* op_reg_data; in RunShapeInference() local
36 TF_RETURN_IF_ERROR(lib_def.LookUp(ndef.op(), &op_reg_data)); in RunShapeInference()
37 if (op_reg_data->shape_inference_fn == nullptr) return Status::OK(); in RunShapeInference()
40 TF_GRAPH_DEF_VERSION, ndef, op_reg_data->op_def, in RunShapeInference()
48 TF_RETURN_IF_ERROR(ic.Run(op_reg_data->shape_inference_fn)); in RunShapeInference()
Dattr_builder.cc58 const OpRegistrationData* op_reg_data = nullptr; in OpDefForOp() local
59 Status s = OpRegistry::Global()->LookUp(op_name, &op_reg_data); in OpDefForOp()
61 *op_def = &op_reg_data->op_def; in OpDefForOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_tf_dialect_op.cc94 mlir::Operation* inst, const tensorflow::OpRegistrationData* op_reg_data, in GetUnregisteredAttrs() argument
96 if (!op_reg_data) { in GetUnregisteredAttrs()
105 for (const auto& attr_def : op_reg_data->op_def.attr()) { in GetUnregisteredAttrs()
122 const tensorflow::OpRegistrationData* op_reg_data, in GetAttributesToIgnore() argument
138 GetUnregisteredAttrs(inst, op_reg_data, &attrs_to_ignore)); in GetAttributesToIgnore()
202 const tensorflow::OpRegistrationData* op_reg_data, in GetAttrValuesFromOperation() argument
208 GetAttributesToIgnore(inst, derived_attrs, op_reg_data, in GetAttrValuesFromOperation()
225 const tensorflow::OpRegistrationData* op_reg_data = in ConvertTFDialectOpToNodeDef() local
227 TF_RETURN_IF_ERROR(GetAttrValuesFromOperation(inst, name, op_reg_data, in ConvertTFDialectOpToNodeDef()
Dexport_tf_dialect_op.h35 const tensorflow::OpRegistrationData* op_reg_data,
/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/ops/
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()
Dfunctional_ops_test.cc36 const OpRegistrationData* op_reg_data; in TEST() local
37 TF_ASSERT_OK(OpRegistry::Global()->LookUp(op.name, &op_reg_data)); in TEST()
39 op.graph_def_version, op.node_def, op_reg_data->op_def, in TEST()
41 TF_ASSERT_OK(c.Run(op_reg_data->shape_inference_fn)); in TEST()
Dmath_ops_test.cc224 const OpRegistrationData* op_reg_data; in TEST() local
225 TF_ASSERT_OK(OpRegistry::Global()->LookUp(op.name, &op_reg_data)); in TEST()
230 CHECK(op_reg_data->shape_inference_fn != nullptr); in TEST()
232 TF_GRAPH_DEF_VERSION, op.node_def, op_reg_data->op_def, in TEST()
236 Status s = c->Run(op_reg_data->shape_inference_fn); in TEST()
/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/core/common_runtime/
Dshape_refiner.cc236 const OpRegistrationData* op_reg_data; in AddNodeInternal() local
237 TF_RETURN_IF_ERROR(ops_registry_->LookUp(node->type_string(), &op_reg_data)); in AddNodeInternal()
238 if (op_reg_data->shape_inference_fn == nullptr && in AddNodeInternal()
249 TF_RETURN_IF_ERROR(RunShapeFn(node, op_reg_data, ec.get(), outer_context)); in AddNodeInternal()
377 const OpRegistrationData* op_reg_data; in UpdateNode() local
378 TF_RETURN_IF_ERROR(ops_registry_->LookUp(node->type_string(), &op_reg_data)); in UpdateNode()
379 if (op_reg_data->shape_inference_fn == nullptr && in UpdateNode()
386 if (!op_reg_data->shape_inference_fn) { in UpdateNode()
391 return RunShapeFn(node, op_reg_data, node_ext_context); in UpdateNode()
648 const OpRegistrationData* op_reg_data, in RunShapeFn() argument
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dshape_inference_utils.cc194 const tensorflow::OpRegistrationData* op_reg_data = in InferReturnTypeComponentsForTFOp() local
196 if (!op_reg_data) { in InferReturnTypeComponentsForTFOp()
201 if (!op_reg_data->shape_inference_fn) { in InferReturnTypeComponentsForTFOp()
212 op, op_name, op_reg_data, /*ignore_unregistered_attrs=*/true, in InferReturnTypeComponentsForTFOp()
259 op_reg_data->op_def, input_shapes, input_tensors, in InferReturnTypeComponentsForTFOp()
261 auto status = c.Run(op_reg_data->shape_inference_fn); in InferReturnTypeComponentsForTFOp()
296 auto status = c.Run(op_reg_data->shape_inference_fn); in InferReturnTypeComponentsForTFOp()
/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/compiler/mlir/lite/utils/
Dtftext_utils_test.cc33 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register() argument
34 op_reg_data->op_def.set_name(op_name); in Register()
/external/tensorflow/tensorflow/lite/tools/
Dlist_flex_ops.cc54 const tensorflow::OpRegistrationData* op_reg_data; in FindTensorflowKernelClass() local
56 tensorflow::OpRegistry::Global()->LookUp(node_def->op(), &op_reg_data); in FindTensorflowKernelClass()
60 AddDefaultsToNodeDef(op_reg_data->op_def, node_def); in FindTensorflowKernelClass()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc469 const tensorflow::OpRegistrationData* op_reg_data; in TF_OpIsStateful() local
471 tensorflow::OpRegistry::Global()->LookUp(op_type, &op_reg_data); in TF_OpIsStateful()
475 return op_reg_data->op_def.is_stateful(); in TF_OpIsStateful()
622 const tensorflow::OpRegistrationData* op_reg_data; in TFE_InferShapes() local
624 tensorflow::OpRegistry::Global()->LookUp(node_def.op(), &op_reg_data); in TFE_InferShapes()
648 InferenceContext c(TF_GRAPH_DEF_VERSION, node_def, op_reg_data->op_def, in TFE_InferShapes()
674 if (op_reg_data->shape_inference_fn == nullptr) { in TFE_InferShapes()
681 status->status = c.Run(op_reg_data->shape_inference_fn); in TFE_InferShapes()
/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/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc63 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomOps() argument
64 *op_reg_data = tensorflow::OpRegistrationData(opdef); in RegisterCustomOps()

123