HomeSort by relevance Sort by last modified time
    Searched refs:IdentityNOp (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
identity_n_op.cc 26 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);
identity_n_op.h 23 class IdentityNOp : public OpKernel {
25 explicit IdentityNOp(OpKernelConstruction* context) : OpKernel(context) {}

Completed in 3636 milliseconds