| /external/tensorflow/tensorflow/core/ops/ |
| 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()
|
| D | shape_function_test.cc | 28 for (const OpRegistrationData& op_reg_data : op_data) { in TEST() local
|
| D | math_ops_test.cc | 203 const OpRegistrationData* op_reg_data; in TEST() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op.cc | 40 const OpRegistrationData* op_reg_data = nullptr; in LookUpOpDef() local
221 std::unique_ptr<OpRegistrationData> op_reg_data(new OpRegistrationData); in RegisterAlreadyLocked() local
253 auto* op_reg_data = new OpRegistrationData(); in OpListOpRegistry() local
288 [wrapper](OpRegistrationData* op_reg_data) -> Status { in OpDefBuilderReceiver()
|
| 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
|
| 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() local
53 OpRegistrationData op_reg_data; in TEST() local
73 OpRegistrationData op_reg_data; in TEST() local
101 OpRegistrationData op_reg_data; in TEST() local
217 OpRegistrationData op_reg_data; in TEST() local
320 OpRegistrationData op_reg_data; in TEST() local
|
| D | shape_inference_testutil.cc | 32 const OpRegistrationData* op_reg_data; in InferShapes() local
|
| D | op_def_util_test.cc | 46 OpRegistrationData op_reg_data; in TestBuilder() local
|
| D | graph_def_util_test.cc | 32 OpRegistrationData op_reg_data; in FinalizeOpDef() local
|
| D | dataset.cc | 277 const OpRegistrationData* op_reg_data = nullptr; in AddFunction() local
|
| D | node_def_util_test.cc | 35 OpRegistrationData op_reg_data; in ToOpDef() local
|
| D | node_def_builder_test.cc | 36 OpRegistrationData op_reg_data; in Op() local
|
| D | shape_inference_test.cc | 33 OpRegistrationData op_reg_data; in MakeOpDefWithLists() local
103 OpRegistrationData op_reg_data; in MakeOpDef() local
1274 OpRegistrationData op_reg_data; in TEST_F() local
|
| /external/tensorflow/tensorflow/python/util/ |
| D | kernel_registry.cc | 35 const tensorflow::OpRegistrationData* op_reg_data; in TryFindKernelClass() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | validate_test.cc | 99 OpRegistrationData op_reg_data; in TEST() local
111 OpRegistrationData op_reg_data; in TEST() local
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | model_analyzer.cc | 82 const OpRegistrationData* op_reg_data; in PrintNodeInfo() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 245 const OpRegistrationData* op_reg_data; in AddNode() local
399 const OpRegistrationData* op_reg_data; in UpdateNode() local
622 const OpRegistrationData* op_reg_data, in RunShapeFn()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.cc | 65 const OpRegistrationData* op_reg_data; in RunCppShapeInferenceImpl() local
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | update_api_def.cc | 141 OpRegistrationData op_reg_data; in ValidateOpDocs() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.cc | 57 const OpRegistrationData* op_reg_data = nullptr; in OpDefForOp() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 49 const tensorflow::OpRegistrationData* op_reg_data = nullptr; in GetNumRetvals() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner.cc | 83 const OpRegistrationData* op_reg_data = nullptr; in IsOutputPortRefValue() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization.cc | 154 const OpRegistrationData* op_reg_data; // Owned by global op registry in GetInputMap() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 225 const tensorflow::OpRegistrationData* op_reg_data; in InitializeNodeDef() local
|