HomeSort by relevance Sort by last modified time
    Searched refs:TensorListStack (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
list_kernels.cu.cc 40 REGISTER_KERNEL_BUILDER(Name("TensorListStack") \
43 TensorListStack<GPUDevice, T>)
list_kernels.h 62 class TensorListStack : public OpKernel {
66 explicit TensorListStack(OpKernelConstruction* c) : OpKernel(c) {
71 ~TensorListStack() {}
list_kernels.cc 478 REGISTER_KERNEL_BUILDER(Name("TensorListStack") \
481 TensorListStack<CPUDevice, T>)

Completed in 80 milliseconds