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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
pack_op.cc 39 class PackOp : public XlaOpKernel {
41 explicit PackOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {
90 REGISTER_XLA_OP(Name("Pack"), PackOp);
  /external/tensorflow/tensorflow/core/kernels/
pack_op.cc 43 class PackOp : public OpKernel {
48 explicit PackOp(OpKernelConstruction* context) : OpKernel(context) {
138 PackOp<CPUDevice, type>)
156 PackOp<GPUDevice, type>)
172 PackOp<CPUDevice, int32>);
180 PackOp<SYCLDevice, type>)
188 PackOp<CPUDevice, int32>);

Completed in 98 milliseconds