Home
last modified time | relevance | path

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

123

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A 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
H A 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 ()()
H A 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
H A Dop_registration_test.cc28 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register()
H A Dnode_properties_test.cc26 OpRegistrationData op_reg_data; in ToOpDef() local
H A Dshape_inference_testutil.cc32 const OpRegistrationData* op_reg_data; in InferShapes() local
H A Dshape_inference_test.cc39 OpRegistrationData op_reg_data; in MakeOpDefWithLists() local
116 OpRegistrationData op_reg_data; in MakeOpDef() local
1354 OpRegistrationData op_reg_data; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dvalidate_test.cc99 OpRegistrationData op_reg_data; in TEST() local
111 OpRegistrationData op_reg_data; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/graph_compactor/
H A Dpass.cc209 const tensorflow::OpRegistrationData *op_reg_data = in removeDefaultValuedAttrs() local
296 const tensorflow::OpRegistrationData *op_reg_data = in addDefaultValuedAttrs() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dshape_function_test.cc28 for (const OpRegistrationData& op_reg_data : op_data) { in TEST() local
H A Dfunctional_ops_test.cc36 const OpRegistrationData* op_reg_data; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dkernel_registry.cc35 const tensorflow::OpRegistrationData* op_reg_data; in TryFindKernelClass() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/
H A Dtf_op_registry.cc31 const tensorflow::OpRegistrationData *op_reg_data = in IsStatefulImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dshape_inference.cc32 const tensorflow::OpRegistrationData* op_reg_data; in RunShapeInference() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dtftext_utils_test.cc34 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register()
/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/
H A Dmodel_analyzer.cc82 const OpRegistrationData* op_reg_data; in PrintNodeInfo() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dlist_flex_ops.cc54 const tensorflow::OpRegistrationData* op_reg_data; in FindTensorflowKernelClass() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
H A Droundtrip.cc100 const OpRegistrationData* op_reg_data; in NormalizeNode() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc478 const tensorflow::OpRegistrationData* op_reg_data; in TF_OpIsStateful() local
629 const tensorflow::OpRegistrationData* op_reg_data; in TFE_InferShapes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
H A Dtf_to_tfjs_json.cc64 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomOps()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Dtf_tfl_flatbuffer_helpers.cc187 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomBuiltinOps()
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/
H A Dupdate_api_def.cc143 OpRegistrationData op_reg_data; in ValidateOpDocs() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops.cc261 const tensorflow::OpRegistrationData *op_reg_data = nullptr; in getRegisteredInterfaceForOp() local

123