Home
last modified time | relevance | path

Searched defs:GatherOp (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dgather_scatter_ops.cc25 class GatherOp : public XlaOpKernel { class
27 explicit GatherOp(OpKernelConstruction* context) : XlaOpKernel(context) { in GatherOp() function in tensorflow::__anon98539b8d0111::GatherOp
H A Dgather_op.cc238 class GatherOp : public XlaOpKernel { class
240 explicit GatherOp(OpKernelConstruction* context) : XlaOpKernel(context) { in GatherOp() function in tensorflow::GatherOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dgather_op.cc38 class GatherOp : public OpKernel { class
45 explicit GatherOp(OpKernelConstruction* c) : OpKernel(c) { in GatherOp() function in tensorflow::GatherOp