Home
last modified time | relevance | path

Searched defs:op_reg_data (Results 1 – 25 of 50) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_tf_dialect_op.cc94 mlir::Operation* inst, const tensorflow::OpRegistrationData* op_reg_data, in GetUnregisteredAttrs()
122 const tensorflow::OpRegistrationData* op_reg_data, in GetAttributesToIgnore()
202 const tensorflow::OpRegistrationData* op_reg_data, in GetAttrValuesFromOperation()
225 const tensorflow::OpRegistrationData* op_reg_data = in ConvertTFDialectOpToNodeDef() local
/external/tensorflow/tensorflow/core/framework/
Dop.cc46 const OpRegistrationData* op_reg_data = nullptr; in LookUpOpDef() local
235 std::unique_ptr<OpRegistrationData> op_reg_data(new OpRegistrationData); in RegisterAlreadyLocked() local
267 auto* op_reg_data = new OpRegistrationData(); in OpListOpRegistry() local
297 std::move(builder_)](OpRegistrationData* op_reg_data) -> Status { in operator ()()
Dop_def_builder_test.cc45 OpRegistrationData op_reg_data; in ExpectSuccess() local
65 OpRegistrationData op_reg_data; in ExpectOrdered() local
78 OpRegistrationData op_reg_data; in ExpectFailure() local
630 OpRegistrationData op_reg_data; in TEST_F() local
Dop_registration_test.cc28 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register()
Dcommon_shape_fns_test.cc38 OpRegistrationData op_reg_data; in MakeOpDef() local
53 OpRegistrationData op_reg_data; in TEST() local
73 OpRegistrationData op_reg_data; in TEST() local
101 OpRegistrationData op_reg_data; in TEST() local
408 OpRegistrationData op_reg_data; in TEST() local
538 OpRegistrationData op_reg_data; in TEST() local
Dnode_properties_test.cc26 OpRegistrationData op_reg_data; in ToOpDef() local
Dshape_inference_testutil.cc32 const OpRegistrationData* op_reg_data; in InferShapes() local
Dgraph_def_util.cc173 const OpRegistrationData* op_reg_data = nullptr; in StripDefaultAttributes() local
Dop_def_util_test.cc47 OpRegistrationData op_reg_data; in TestBuilder() local
/external/tensorflow/tensorflow/core/ops/
Dshape_function_test.cc28 for (const OpRegistrationData& op_reg_data : op_data) { in TEST() local
Dfunctional_ops_test.cc36 const OpRegistrationData* op_reg_data; in TEST() local
/external/tensorflow/tensorflow/python/util/
Dkernel_registry.cc35 const tensorflow::OpRegistrationData* op_reg_data; in TryFindKernelClass() local
/external/tensorflow/tensorflow/core/graph/
Dvalidate_test.cc99 OpRegistrationData op_reg_data; in TEST() local
111 OpRegistrationData op_reg_data; in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dshape_inference.cc32 const tensorflow::OpRegistrationData* op_reg_data; in RunShapeInference() local
Dattr_builder.cc58 const OpRegistrationData* op_reg_data = nullptr; in OpDefForOp() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dtftext_utils_test.cc33 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register()
/external/tensorflow/tensorflow/python/grappler/
Dmodel_analyzer.cc82 const OpRegistrationData* op_reg_data; in PrintNodeInfo() local
/external/tensorflow/tensorflow/lite/tools/
Dlist_flex_ops.cc54 const tensorflow::OpRegistrationData* op_reg_data; in FindTensorflowKernelClass() local
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc236 const OpRegistrationData* op_reg_data; in AddNodeInternal() local
377 const OpRegistrationData* op_reg_data; in UpdateNode() local
648 const OpRegistrationData* op_reg_data, in RunShapeFn()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc63 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomOps()
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dtf_tfl_flatbuffer_helpers.cc174 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomBuiltinOps()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc88 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterExtraTfOpDefs()
/external/tensorflow/tensorflow/core/api_def/
Dupdate_api_def.cc141 OpRegistrationData op_reg_data; in ValidateOpDocs() local
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel_test.cc357 tensorflow::OpRegistrationData op_reg_data; in MakeOpDef() local
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_python_api.cc357 tensorflow::OpRegistrationData* op_reg_data) -> tensorflow::Status { in RegisterCustomOpdefs()

12