OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:StackCloseOp
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
stack_ops.cc
235
class
StackCloseOp
: public XlaOpKernel {
237
explicit
StackCloseOp
(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
244
TF_DISALLOW_COPY_AND_ASSIGN(
StackCloseOp
);
247
REGISTER_XLA_OP(Name("StackCloseV2"),
StackCloseOp
);
/external/tensorflow/tensorflow/core/kernels/
stack_ops.cc
503
class
StackCloseOp
: public OpKernel {
505
explicit
StackCloseOp
(OpKernelConstruction* context) : OpKernel(context) {}
517
REGISTER_KERNEL_BUILDER(Name("StackClose").Device(DEVICE_CPU),
StackCloseOp
);
519
Name("StackClose").Device(DEVICE_GPU).HostMemory("handle"),
StackCloseOp
);
520
REGISTER_KERNEL_BUILDER(Name("StackCloseV2").Device(DEVICE_CPU),
StackCloseOp
);
522
Name("StackCloseV2").Device(DEVICE_GPU).HostMemory("handle"),
StackCloseOp
);
525
Name("StackClose").Device(DEVICE_SYCL).HostMemory("handle"),
StackCloseOp
);
528
StackCloseOp
);
Completed in 84 milliseconds