Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsharding_util_ops.cc269 class ReadVariableXlaSplitNDOp : public XlaSplitNDBaseOp<true> { class
271 explicit ReadVariableXlaSplitNDOp(OpKernelConstruction* ctx) in ReadVariableXlaSplitNDOp() function in tensorflow::__anone83cea6a0111::ReadVariableXlaSplitNDOp
/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops.cc611 class ReadVariableXlaSplitNDOp : public XlaSplitNDBaseOp<Device, T> { class
613 explicit ReadVariableXlaSplitNDOp(OpKernelConstruction* ctx) in ReadVariableXlaSplitNDOp() function in tensorflow::__anonda0888800111::ReadVariableXlaSplitNDOp