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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
identity_op.cc 22 class IdentityOp : public XlaOpKernel {
24 explicit IdentityOp(OpKernelConstruction* context) : XlaOpKernel(context) {}
33 TF_DISALLOW_COPY_AND_ASSIGN(IdentityOp);
38 REGISTER_XLA_OP(Name("Identity").CompilationOnly(), IdentityOp);
40 REGISTER_XLA_OP(Name("IdentityN").CompilationOnly(), IdentityOp);
41 REGISTER_XLA_OP(Name("PreventGradient"), IdentityOp);
42 REGISTER_XLA_OP(Name("StopGradient"), IdentityOp);
43 REGISTER_XLA_OP(Name("Snapshot"), IdentityOp);
  /external/tensorflow/tensorflow/core/kernels/
identity_op.cc 26 REGISTER_KERNEL_BUILDER(Name("Identity").Device(DEVICE_CPU), IdentityOp);
29 REGISTER_KERNEL_BUILDER(Name("StopGradient").Device(DEVICE_CPU), IdentityOp);
32 REGISTER_KERNEL_BUILDER(Name("PreventGradient").Device(DEVICE_CPU), IdentityOp);
37 IdentityOp);
39 REGISTER_KERNEL_BUILDER(Name("RefIdentity").Device(DEVICE_CPU), IdentityOp);
44 IdentityOp);
46 IdentityOp);
52 IdentityOp); \
55 IdentityOp); \
58 IdentityOp); \
    [all...]
identity_op.h 23 class IdentityOp : public OpKernel {
25 explicit IdentityOp(OpKernelConstruction* context) : OpKernel(context) {}
dynamic_partition_op_gpu.cu.cc 112 struct IdentityOp {
121 : public TransformOutputIterator<int32, int32, IdentityOp> {
131 BoundedReference(int32* ptr, int32* base, IdentityOp op, int32 limit)
146 IdentityOp op,
151 BoundedOutputIterator(int32* ptr, int32* base, IdentityOp op, int32 size)
393 IdentityOp id_op;
  /external/tensorflow/tensorflow/compiler/jit/
xla_device_ops.h 65 Name("Identity").Device(DEVICE).TypeConstraint("T", TYPES), IdentityOp); \

Completed in 325 milliseconds