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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
select_op.cc 28 class SelectOp : public XlaOpKernel {
30 explicit SelectOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
83 TF_DISALLOW_COPY_AND_ASSIGN(SelectOp);
86 REGISTER_XLA_OP(Name("Select"), SelectOp);
  /external/tensorflow/tensorflow/core/kernels/
cwise_op_select.cc 36 class SelectOp : public OpKernel {
38 explicit SelectOp(OpKernelConstruction* context) : OpKernel(context) {}
146 TF_DISALLOW_COPY_AND_ASSIGN(SelectOp);
152 SelectOp<CPUDevice, type>);
162 SelectOp<GPUDevice, type>);
181 SelectOp<SYCLDevice, type>);

Completed in 73 milliseconds