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()
|