Searched defs:AllocateOutputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ | ||
| D | sparse_split_op_gpu.cu.cc | 343 Status AllocateOutputs(OpKernelContext* context, int num_split, int rank, in AllocateOutputs() function |
| D | dynamic_partition_op_gpu.cu.cc | 207 void AllocateOutputs(OpKernelContext* c, const Tensor* data, in AllocateOutputs() function in tensorflow::DynamicPartitionOpGPU |
| D | cudnn_rnn_ops.cc | 1587 Status AllocateOutputs(OpKernelContext* context, in AllocateOutputs() function in tensorflow::CudnnRNNForwardOp |