Searched refs:GatherOp (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 149 class GatherOp : public XlaOpKernel { class
151 explicit GatherOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in GatherOp() function in tensorflow::GatherOp
192 TF_DISALLOW_COPY_AND_ASSIGN(GatherOp);
195 REGISTER_XLA_OP(Name("Gather"), GatherOp);
196 REGISTER_XLA_OP(Name("GatherV2").CompileTimeConstantInput("axis"), GatherOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_op.cc | 35 class GatherOp : public OpKernel { class
42 explicit GatherOp(OpKernelConstruction* c) : OpKernel(c) {} in GatherOp() function in tensorflow::GatherOp
127 GatherOp<dev##Device, type, index_type>); \
133 GatherOp<dev##Device, type, index_type>)
|
| D | reduction_gpu_kernels.cu.h | 476 struct GatherOp {
477 __host__ __device__ GatherOp(const int& extent_x, const int& extent_y,
743 GatherOp gather_op(extent_x, extent_y, extent_z, false);
744 typedef cub::TransformInputIterator<int, GatherOp,
|
| /external/swiftshader/third_party/LLVM/lib/Target/CellSPU/ |
| D | SPUSchedule.td | 26 def GatherOp : InstrItinClass; // ODD_UNIT
47 InstrItinData<GatherOp , [InstrStage<4, [ODD_UNIT]>]>,
|
| D | SPUInstrInfo.td | 1032 RRForm_1<0b01001101100, OOL, IOL, "gbb\t$rT, $rA", GatherOp, pattern>;
1057 RRForm_1<0b10001101100, OOL, IOL, "gbh\t$rT, $rA", GatherOp,
1083 RRForm_1<0b00001101100, OOL, IOL, "gb\t$rT, $rA", GatherOp,
|