Searched defs:Roll (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | roll_op_gpu.cu.cc | 55 struct Roll<GPUDevice, T> { struct
56 void operator()(const OpKernelContext* context, const int64 num_elements, in operator ()()
|
| D | roll_op.h | 28 struct Roll { struct
|
| D | roll_op.cc | 306 struct Roll<CPUDevice, T> { struct
307 void operator()(const OpKernelContext* context, const int64 num_elements, in operator ()()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 264 Node* Roll(Graph* g, Node* input, Node* shift, Node* axis) { in Roll() function
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 17647 func Roll(scope *Scope, input tf.Output, shift tf.Output, axis tf.Output) (output tf.Output) { func
|