Searched refs:TF_NewKernelBuilder (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op.cc | 143 auto* builder = TF_NewKernelBuilder("Bitcast", tensorflow::DEVICE_CPU, in RegisterBitcastOpKernel()
153 auto* builder = TF_NewKernelBuilder("Bitcast", tensorflow::DEVICE_GPU, in RegisterBitcastOpKernel()
|
| D | merge_summary_op.cc | 104 auto* builder = TF_NewKernelBuilder( in RegisterMergeSummaryOpKernel()
|
| D | histogram_summary_op.cc | 132 auto* builder = TF_NewKernelBuilder( in RegisterHistogramSummaryOpKernel()
|
| D | summary_op.cc | 139 auto* builder = TF_NewKernelBuilder( in RegisterScalarSummaryOpKernel()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 135 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
212 TF_KernelBuilder* builder = TF_NewKernelBuilder( in CreateAndCallKernelWithAttr()
548 TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, nullptr, in TEST()
603 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
647 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
678 TF_KernelBuilder* builder = TF_NewKernelBuilder( in SetupOp()
945 TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, nullptr, in TEST_F()
|
| D | c_test.c | 84 TF_NewKernelBuilder("SomeOp", "SomeDevice", &create, &compute, NULL); in main()
|
| D | kernels.h | 102 TF_CAPI_EXPORT extern TF_KernelBuilder* TF_NewKernelBuilder(
|
| D | kernels.cc | 48 TF_KernelBuilder* TF_NewKernelBuilder( in TF_NewKernelBuilder() function
|
| /external/tensorflow/tensorflow/lite/toco/python/ |
| D | toco_python_api.cc | 370 TF_NewKernelBuilder(op_name, device_name, /*create_func=*/nullptr, in RegisterCustomOpdefs()
|