Home
last modified time | relevance | path

Searched refs:TF_NewKernelBuilder (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op.cc142 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/
Dkernels_test.cc99 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
186 TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, nullptr, in TEST()
Dkernels.h70 TF_CAPI_EXPORT extern TF_KernelBuilder* TF_NewKernelBuilder(
Dc_test.c88 TF_NewKernelBuilder("SomeOp", "SomeDevice", &create, &compute, NULL); in main()
Dkernels.cc36 TF_KernelBuilder* TF_NewKernelBuilder( in TF_NewKernelBuilder() function