Searched refs:ConcatBaseOp (Results 1 – 2 of 2) sorted by relevance
41 class ConcatBaseOp : public XlaOpKernel { class43 ConcatBaseOp(OpKernelConstruction* c, int axis_index) in ConcatBaseOp() function in tensorflow::__anon8bb65f0c0111::ConcatBaseOp100 class ConcatOp : public ConcatBaseOp {103 : ConcatBaseOp(c, /* axis_index */ 0) {} in ConcatOp()108 class ConcatV2Op : public ConcatBaseOp {111 : ConcatBaseOp(c, /* axis_index */ c->num_inputs() - 1) {} in ConcatV2Op()
43 class ConcatBaseOp : public OpKernel { class48 explicit ConcatBaseOp(OpKernelConstruction* c) in ConcatBaseOp() function in tensorflow::ConcatBaseOp180 using ConcatOp = ConcatBaseOp<Device, T, NAME_IS_CONCAT_DIM>;182 using ConcatV2Op = ConcatBaseOp<Device, T, NAME_IS_AXIS>;