Searched refs:TF_NewKernelBuilder (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 123 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
210 TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, nullptr, in TEST()
265 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
309 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
340 TF_KernelBuilder* builder = TF_NewKernelBuilder( in SetupOp()
|
| D | c_test.c | 84 TF_NewKernelBuilder("SomeOp", "SomeDevice", &create, &compute, NULL); in main()
|
| D | kernels.h | 94 TF_CAPI_EXPORT extern TF_KernelBuilder* TF_NewKernelBuilder(
|
| D | kernels.cc | 41 TF_KernelBuilder* TF_NewKernelBuilder( in TF_NewKernelBuilder() function
|
| /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()
|