| /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 | 40 const OpRegistrationData* op_reg_data = nullptr; in LookUpOpDef()
62 const OpRegistrationData** op_reg_data) const { in LookUp()
66 if (const OpRegistrationData* res = in LookUp()
77 const OpRegistrationData** op_reg_data) const { in LookUpSlow()
79 const OpRegistrationData* res = nullptr; in LookUpSlow()
134 std::vector<OpRegistrationData>* op_data) { in GetOpRegistrationData()
156 std::vector<std::pair<string, const OpRegistrationData*>> sorted( in Export()
221 std::unique_ptr<OpRegistrationData> op_reg_data(new OpRegistrationData); in RegisterAlreadyLocked()
253 auto* op_reg_data = new OpRegistrationData(); in OpListOpRegistry()
264 const OpRegistrationData** op_reg_data) const { in LookUp()
[all …]
|
| 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,
152 Status Finalize(OpRegistrationData* op_reg_data) const;
164 OpRegistrationData op_reg_data_;
|
| D | op.h | 49 const OpRegistrationData** op_reg_data) const = 0;
67 typedef std::function<Status(OpRegistrationData*)> OpRegistrationDataFactory;
75 const OpRegistrationData** op_reg_data) const override;
93 void GetOpRegistrationData(std::vector<OpRegistrationData>* op_data);
148 const OpRegistrationData** op_reg_data) const;
154 mutable std::unordered_map<string, const OpRegistrationData*> registry_
173 const OpRegistrationData** op_reg_data) const override;
177 std::unordered_map<string, const OpRegistrationData*> index_;
|
| 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()
217 OpRegistrationData op_reg_data; in TEST()
320 OpRegistrationData op_reg_data; in TEST()
|
| D | function.h | 390 const OpRegistrationData** op_reg_data) const override
449 OpRegistrationData op_registration_data;
|
| D | shape_inference_testutil.cc | 32 const OpRegistrationData* op_reg_data; in InferShapes()
|
| D | graph_def_util_test.cc | 32 OpRegistrationData op_reg_data; in FinalizeOpDef()
|
| D | op_def_util_test.cc | 46 OpRegistrationData op_reg_data; in TestBuilder()
|
| D | dataset.cc | 277 const OpRegistrationData* op_reg_data = nullptr; in AddFunction()
|
| /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 | tpu_embedding_ops.cc | 78 OpRegistrationData* op_reg_data) { in RegisterPerTableLoadOpsForAlgorithmBody()
194 OpRegistrationData* op_reg_data) { in RegisterPerTableRetrieveOpsForAlgorithmBody()
292 [alg](OpRegistrationData* op_reg_data) -> Status { in RegisterPerTableLoadAndRetrieveOps()
300 [alg](OpRegistrationData* op_reg_data) -> Status { in RegisterPerTableLoadAndRetrieveOps()
309 [alg](OpRegistrationData* op_reg_data) -> Status { in RegisterPerTableLoadAndRetrieveOps()
317 [alg](OpRegistrationData* op_reg_data) -> Status { in RegisterPerTableLoadAndRetrieveOps()
|
| /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/python/grappler/ |
| D | model_analyzer.cc | 82 const OpRegistrationData* op_reg_data; in PrintNodeInfo()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 245 const OpRegistrationData* op_reg_data; in AddNode()
399 const OpRegistrationData* op_reg_data; in UpdateNode()
622 const OpRegistrationData* op_reg_data, in RunShapeFn()
|
| D | shape_refiner.h | 260 Status RunShapeFn(const Node* node, const OpRegistrationData* op_reg_data,
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 39 Status RegisterFunctionBodyOutputs(const OpRegistrationData& registration, in RegisterFunctionBodyOutputs()
53 const OpRegistrationData* registration; in RegisterFunctionBodyOutputs()
584 const OpRegistrationData* registration; in MakeGrapplerFunctionItem()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.cc | 65 const OpRegistrationData* op_reg_data; in RunCppShapeInferenceImpl()
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | update_api_def.cc | 141 OpRegistrationData op_reg_data; in ValidateOpDocs()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 137 const tensorflow::OpRegistrationData** op_data);
|
| D | attr_builder.cc | 57 const OpRegistrationData* op_reg_data = nullptr; in OpDefForOp()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 49 const tensorflow::OpRegistrationData* op_reg_data = nullptr; in GetNumRetvals()
|