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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
select_op.cc 29 class SelectOp : public XlaOpKernel {
31 explicit SelectOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
82 TF_DISALLOW_COPY_AND_ASSIGN(SelectOp);
85 REGISTER_XLA_OP(Name("Select"), SelectOp);
  /external/tensorflow/tensorflow/core/kernels/
cwise_op_select.cc 42 class SelectOp : public OpKernel {
44 explicit SelectOp(OpKernelConstruction* context) : OpKernel(context) {}
144 TF_DISALLOW_COPY_AND_ASSIGN(SelectOp);
150 SelectOp<CPUDevice, type>);
160 SelectOp<GPUDevice, type>);
180 SelectOp<SYCLDevice, type>);

Completed in 83 milliseconds