Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_tf_dialect_op.cc93 mlir::Operation* inst, const tensorflow::OpRegistrationData* op_reg_data, in GetUnregisteredAttrs()
121 const tensorflow::OpRegistrationData* op_reg_data, in GetAttributesToIgnore()
200 const tensorflow::OpRegistrationData* op_reg_data, in GetAttrValuesFromOperation()
242 const tensorflow::OpRegistrationData* op_reg_data = in ConvertTFDialectOpToNodeDef() local
/external/tensorflow/tensorflow/core/framework/
Dop.cc47 const OpRegistrationData* op_reg_data = nullptr; in LookUpOpDef() local
236 std::unique_ptr<OpRegistrationData> op_reg_data(new OpRegistrationData); in RegisterAlreadyLocked() local
268 auto* op_reg_data = new OpRegistrationData(); in OpListOpRegistry() local
298 std::move(builder_)](OpRegistrationData* op_reg_data) -> Status { in operator ()()
Dcommon_shape_fns_test.cc39 OpRegistrationData op_reg_data; in MakeOpDef() local
54 OpRegistrationData op_reg_data; in TEST() local
74 OpRegistrationData op_reg_data; in TEST() local
102 OpRegistrationData op_reg_data; in TEST() local
410 OpRegistrationData op_reg_data; in TEST() local
540 OpRegistrationData op_reg_data; in TEST() local
1493 OpRegistrationData op_reg_data; in TEST() local
1522 OpRegistrationData op_reg_data; in TEST() local
1549 OpRegistrationData op_reg_data; in TEST() local
1577 OpRegistrationData op_reg_data; in TEST() local
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()
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
/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/ir/
Dtf_op_registry.cc31 const tensorflow::OpRegistrationData *op_reg_data = in IsStatefulImpl() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dshape_inference.cc32 const tensorflow::OpRegistrationData* op_reg_data; in RunShapeInference() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dtftext_utils_test.cc34 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register()
/external/tensorflow/tensorflow/core/transforms/graph_compactor/
Dpass.cc209 const tensorflow::OpRegistrationData *op_reg_data = in removeDefaultValuedAttrs() local
296 const tensorflow::OpRegistrationData *op_reg_data = in addDefaultValuedAttrs() local
/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.cc252 const OpRegistrationData* op_reg_data; in AddNodeInternal() local
393 const OpRegistrationData* op_reg_data; in UpdateNode() local
669 const OpRegistrationData* op_reg_data, in RunShapeFn()
/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
Droundtrip.cc100 const OpRegistrationData* op_reg_data; in NormalizeNode() local
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc64 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomOps()
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dtf_tfl_flatbuffer_helpers.cc187 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomBuiltinOps()
/external/tensorflow/tensorflow/core/api_def/
Dupdate_api_def.cc143 OpRegistrationData op_reg_data; in ValidateOpDocs() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc98 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterExtraTfOpDefs()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.cc261 const tensorflow::OpRegistrationData *op_reg_data = nullptr; in getRegisteredInterfaceForOp() local

123