Searched refs:IdentityOp (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | 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 …]
|
| D | identity_op.h | 23 class IdentityOp : public OpKernel {
25 explicit IdentityOp(OpKernelConstruction* context) : OpKernel(context) {} in IdentityOp() function
|
| D | dynamic_partition_op_gpu.cu.cc | 113 struct IdentityOp { struct
122 : public TransformOutputIterator<int32, int32, IdentityOp> {
132 BoundedReference(int32* ptr, int32* base, IdentityOp op, int32 limit) in BoundedReference()
147 IdentityOp op, in BoundedOutputIterator()
152 BoundedOutputIterator(int32* ptr, int32* base, IdentityOp op, int32 size) in BoundedOutputIterator()
394 IdentityOp id_op; in CountAndSortParts()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | identity_op.cc | 22 class IdentityOp : public XlaOpKernel { class
24 explicit IdentityOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in IdentityOp() function in tensorflow::__anonb0a82ca70111::IdentityOp
36 TF_DISALLOW_COPY_AND_ASSIGN(IdentityOp);
43 IdentityOp);
48 IdentityOp);
49 REGISTER_XLA_OP(Name("PlaceholderWithDefault"), IdentityOp);
50 REGISTER_XLA_OP(Name("PreventGradient"), IdentityOp);
51 REGISTER_XLA_OP(Name("StopGradient"), IdentityOp);
52 REGISTER_XLA_OP(Name("Snapshot"), IdentityOp);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 100 Name("Identity").Device(DEVICE).TypeConstraint("T", TYPES), IdentityOp); \
103 IdentityOp); \
106 IdentityOp); \
112 IdentityOp); \
|