Home
last modified time | relevance | path

Searched refs:ConcatOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconcat_op.cc175 using ConcatOp = ConcatBaseOp<Device, T, NAME_IS_CONCAT_DIM>; typedef
184 ConcatOp<CPUDevice, type>) \
207 ConcatOp<GPUDevice, type>) \
232 ConcatOp<CPUDevice, int32>);
249 ConcatOp<SYCLDevice, type>) \
264 ConcatOp<CPUDevice, int32>);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconcat_op.cc100 class ConcatOp : public ConcatBaseOp { class
102 explicit ConcatOp(OpKernelConstruction* c) in ConcatOp() function in tensorflow::__anon01eae5b50111::ConcatOp
115 ConcatOp);