Home
last modified time | relevance | path

Searched refs:IdentityNOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Didentity_n_op.cc26 REGISTER_KERNEL_BUILDER(Name("IdentityN").Device(DEVICE_CPU), IdentityNOp);
29 REGISTER_KERNEL_BUILDER(Name("IdentityN").Device(DEVICE_SYCL), IdentityNOp);
32 REGISTER_KERNEL_BUILDER(Name("IdentityN").Device(DEVICE_GPU), IdentityNOp);
Didentity_n_op.h23 class IdentityNOp : public OpKernel {
25 explicit IdentityNOp(OpKernelConstruction* context) : OpKernel(context) {} in IdentityNOp() function
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h113 REGISTER_KERNEL_BUILDER(Name("IdentityN").Device(DEVICE), IdentityNOp); \