Searched refs:TF_NewKernelBuilder (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/c/kernels/ |
D | bitcast_op.cc | 142 auto* builder = TF_NewKernelBuilder("Bitcast", tensorflow::DEVICE_CPU, in RegisterBitcastOp() 152 auto* builder = TF_NewKernelBuilder("Bitcast", tensorflow::DEVICE_GPU, in RegisterBitcastOp()
|
/external/tensorflow/tensorflow/c/ |
D | kernels_test.cc | 99 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST() 186 TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, nullptr, in TEST()
|
D | kernels.h | 70 TF_CAPI_EXPORT extern TF_KernelBuilder* TF_NewKernelBuilder(
|
D | c_test.c | 88 TF_NewKernelBuilder("SomeOp", "SomeDevice", &create, &compute, NULL); in main()
|
D | kernels.cc | 36 TF_KernelBuilder* TF_NewKernelBuilder( in TF_NewKernelBuilder() function
|