Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsharding_util_ops.cc129 class XlaSplitNDBaseOp : public XlaOpKernel { class
131 explicit XlaSplitNDBaseOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in XlaSplitNDBaseOp() function in tensorflow::__anone83cea6a0111::XlaSplitNDBaseOp
/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops.cc477 class XlaSplitNDBaseOp : public XlaSplitNDShared { class
479 explicit XlaSplitNDBaseOp(OpKernelConstruction* ctx) in XlaSplitNDBaseOp() function in tensorflow::__anonda0888800111::XlaSplitNDBaseOp