Searched refs:TF_RegisterKernelBuilder (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op.cc | 146 TF_RegisterKernelBuilder("BitcastOp", builder, status); in RegisterBitcastOpKernel()
156 TF_RegisterKernelBuilder("BitcastOp", builder, status); in RegisterBitcastOpKernel()
|
| D | merge_summary_op.cc | 107 TF_RegisterKernelBuilder("MergeSummary", builder, status); in RegisterMergeSummaryOpKernel()
|
| D | histogram_summary_op.cc | 139 TF_RegisterKernelBuilder("HistogramSummary", builder, status); in RegisterHistogramSummaryOpKernel()
|
| D | summary_op.cc | 146 TF_RegisterKernelBuilder("ScalarSummary", builder, status); in RegisterScalarSummaryOpKernel()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 140 TF_RegisterKernelBuilder(node_name, builder, status); in TEST()
216 TF_RegisterKernelBuilder("FakeNode", builder, status); in CreateAndCallKernelWithAttr()
553 TF_RegisterKernelBuilder(node_name, builder, status); in TEST()
608 TF_RegisterKernelBuilder(node_name, builder, status); in TEST()
652 TF_RegisterKernelBuilder(node_name, builder, status); in TEST()
681 TF_RegisterKernelBuilder(node_name, builder, status); in SetupOp()
950 TF_RegisterKernelBuilder(node_name, builder, status); in TEST_F()
|
| D | c_test.c | 85 TF_RegisterKernelBuilder("someKernel", b, status); in main()
|
| D | kernels.h | 126 TF_CAPI_EXPORT extern void TF_RegisterKernelBuilder(const char* kernel_name,
|
| D | kernels.cc | 183 void TF_RegisterKernelBuilder(const char* name, TF_KernelBuilder* builder, in TF_RegisterKernelBuilder() function
|
| /external/tensorflow/tensorflow/lite/toco/python/ |
| D | toco_python_api.cc | 374 TF_RegisterKernelBuilder(node_name, builder, status); in RegisterCustomOpdefs()
|