| /external/tensorflow/tensorflow/core/framework/ |
| D | op_compatibility_test.cc | 222 OpRegistrationData old_op; in TEST_F()
235 OpRegistrationData old_op; in TEST_F()
253 OpRegistrationData old_op; in TEST_F()
271 OpRegistrationData old_op; in TEST_F()
294 OpRegistrationData old_op; in TEST_F()
316 OpRegistrationData old_op; in TEST_F()
339 OpRegistrationData old_op; in TEST_F()
361 OpRegistrationData old_op; in TEST_F()
383 OpRegistrationData old_op; in TEST_F()
409 OpRegistrationData old_op; in TEST_F()
[all …]
|
| D | op.cc | 46 const OpRegistrationData* op_reg_data = nullptr; in LookUpOpDef()
85 const OpRegistrationData** op_reg_data) const { in LookUp()
90 const OpRegistrationData* OpRegistry::LookUp(const string& op_type_name) const { in LookUp()
94 if (const OpRegistrationData* res = in LookUp()
103 const OpRegistrationData* OpRegistry::LookUpSlow( in LookUpSlow()
105 const OpRegistrationData* res = nullptr; in LookUpSlow()
148 std::vector<OpRegistrationData>* op_data) { in GetOpRegistrationData()
170 std::vector<std::pair<string, const OpRegistrationData*>> sorted( in Export()
235 std::unique_ptr<OpRegistrationData> op_reg_data(new OpRegistrationData); in RegisterAlreadyLocked()
267 auto* op_reg_data = new OpRegistrationData(); in OpListOpRegistry()
[all …]
|
| D | op.h | 49 const OpRegistrationData** op_reg_data) const = 0;
68 typedef std::function<Status(OpRegistrationData*)> OpRegistrationDataFactory;
76 const OpRegistrationData** op_reg_data) const override;
79 const OpRegistrationData* LookUp(const std::string& op_type_name) const;
97 void GetOpRegistrationData(std::vector<OpRegistrationData>* op_data);
157 const OpRegistrationData* LookUpSlow(const std::string& op_type_name) const;
163 mutable std::unordered_map<string, const OpRegistrationData*> registry_
184 const OpRegistrationData** op_reg_data) const override;
187 const OpRegistrationData* LookUp(const std::string& op_type_name) const;
191 std::unordered_map<string, const OpRegistrationData*> index_;
|
| D | op_def_builder.h | 39 struct OpRegistrationData { struct
41 OpRegistrationData() {} in OpRegistrationData() argument
42 OpRegistrationData(const OpDef& def) : op_def(def) {} in OpRegistrationData() function
43 OpRegistrationData(const OpDef& def, const OpShapeInferenceFn& fn,
156 Status Finalize(OpRegistrationData* op_reg_data) const;
168 OpRegistrationData op_reg_data_;
|
| D | node_properties_test.cc | 26 OpRegistrationData op_reg_data; in ToOpDef()
44 const OpRegistrationData** op_reg_data) const override { in LookUp()
58 const OpRegistrationData op_reg_;
|
| D | op_def_builder_test.cc | 45 OpRegistrationData op_reg_data; in ExpectSuccess()
65 OpRegistrationData op_reg_data; in ExpectOrdered()
78 OpRegistrationData op_reg_data; in ExpectFailure()
630 OpRegistrationData op_reg_data; in TEST_F()
|
| D | op_registration_test.cc | 28 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register()
|
| D | common_shape_fns_test.cc | 38 OpRegistrationData op_reg_data; in MakeOpDef()
53 OpRegistrationData op_reg_data; in TEST()
73 OpRegistrationData op_reg_data; in TEST()
101 OpRegistrationData op_reg_data; in TEST()
408 OpRegistrationData op_reg_data; in TEST()
538 OpRegistrationData op_reg_data; in TEST()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | shape_function_test.cc | 26 std::vector<OpRegistrationData> op_data; in TEST()
28 for (const OpRegistrationData& op_reg_data : op_data) { in TEST()
|
| D | functional_ops_test.cc | 36 const OpRegistrationData* op_reg_data; in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | tftext_utils_test.cc | 26 using tensorflow::OpRegistrationData;
33 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { in Register()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_tf_dialect_op.cc | 94 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()
|
| D | export_tf_dialect_op.h | 35 const tensorflow::OpRegistrationData* op_reg_data,
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 104 const OpRegistrationData* reg; in TEST()
122 const OpRegistrationData* reg; in TEST()
141 const OpRegistrationData* reg; in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/ |
| D | tf_to_tfjs_json.cc | 63 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomOps()
64 *op_reg_data = tensorflow::OpRegistrationData(opdef); in RegisterCustomOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| D | tf_tfl_flatbuffer_helpers.cc | 169 const OpRegistrationData* op_reg = in RegisterCustomBuiltinOps()
174 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomBuiltinOps()
175 *op_reg_data = tensorflow::OpRegistrationData(opdef); in RegisterCustomBuiltinOps()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | validate_test.cc | 99 OpRegistrationData op_reg_data; in TEST()
111 OpRegistrationData op_reg_data; in TEST()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | kernel_registry.cc | 35 const tensorflow::OpRegistrationData* op_reg_data; in TryFindKernelClass()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | shape_inference.cc | 32 const tensorflow::OpRegistrationData* op_reg_data; in RunShapeInference()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 88 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterExtraTfOpDefs()
89 *op_reg_data = tensorflow::OpRegistrationData(opdef); in RegisterExtraTfOpDefs()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | model_analyzer.cc | 82 const OpRegistrationData* op_reg_data; in PrintNodeInfo()
|
| /external/tensorflow/tensorflow/lite/toco/python/ |
| D | toco_python_api.cc | 357 tensorflow::OpRegistrationData* op_reg_data) -> tensorflow::Status { in RegisterCustomOpdefs()
358 *op_reg_data = tensorflow::OpRegistrationData(opdef); in RegisterCustomOpdefs()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | list_flex_ops.cc | 54 const tensorflow::OpRegistrationData* op_reg_data; in FindTensorflowKernelClass()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 236 const OpRegistrationData* op_reg_data; in AddNodeInternal()
377 const OpRegistrationData* op_reg_data; in UpdateNode()
648 const OpRegistrationData* op_reg_data, in RunShapeFn()
|
| D | shape_refiner.h | 289 Status RunShapeFn(const Node* node, const OpRegistrationData* op_reg_data,
|