/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | common_shape_fns_test.cc | 39 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 D | op.cc | 47 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 D | op_def_builder_test.cc | 45 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 D | op_registration_test.cc | 28 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register()
|
H A D | node_properties_test.cc | 26 OpRegistrationData op_reg_data; in ToOpDef() local
|
H A D | shape_inference_testutil.cc | 32 const OpRegistrationData* op_reg_data; in InferShapes() local
|
H A D | shape_inference_test.cc | 39 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 D | export_tf_dialect_op.cc | 93 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 D | shape_refiner.cc | 252 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 D | validate_test.cc | 99 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 D | pass.cc | 209 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 D | shape_function_test.cc | 28 for (const OpRegistrationData& op_reg_data : op_data) { in TEST() local
|
H A D | functional_ops_test.cc | 36 const OpRegistrationData* op_reg_data; in TEST() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/util/ |
H A D | kernel_registry.cc | 35 const tensorflow::OpRegistrationData* op_reg_data; in TryFindKernelClass() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/ |
H A D | tf_op_registry.cc | 31 const tensorflow::OpRegistrationData *op_reg_data = in IsStatefulImpl() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
H A D | shape_inference.cc | 32 const tensorflow::OpRegistrationData* op_reg_data; in RunShapeInference() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
H A D | tftext_utils_test.cc | 34 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/ |
H A D | model_analyzer.cc | 82 const OpRegistrationData* op_reg_data; in PrintNodeInfo() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/ |
H A D | list_flex_ops.cc | 54 const tensorflow::OpRegistrationData* op_reg_data; in FindTensorflowKernelClass() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/ |
H A D | roundtrip.cc | 100 const OpRegistrationData* op_reg_data; in NormalizeNode() local
|
/aosp_15_r20/external/tensorflow/tensorflow/c/ |
H A D | c_api_experimental.cc | 478 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 D | tf_to_tfjs_json.cc | 64 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomOps()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
H A D | tf_tfl_flatbuffer_helpers.cc | 187 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomBuiltinOps()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/ |
H A D | update_api_def.cc | 143 OpRegistrationData op_reg_data; in ValidateOpDocs() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
H A D | tf_ops.cc | 261 const tensorflow::OpRegistrationData *op_reg_data = nullptr; in getRegisteredInterfaceForOp() local
|