Home
last modified time | relevance | path

Searched defs:DeviceIndexOp (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Ddevice_index_op.cc31 class DeviceIndexOp : public XlaOpKernel { class
33 explicit DeviceIndexOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in DeviceIndexOp() function in tensorflow::__anon5ac8fd510111::DeviceIndexOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfunctional_ops.cc897 class DeviceIndexOp : public OpKernel { class
899 explicit DeviceIndexOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in DeviceIndexOp() function in tensorflow::__anona00ae9de0111::DeviceIndexOp