OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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