Searched defs:AllocateOutputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ | ||
| 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 | 1585 Status AllocateOutputs(OpKernelContext* context, in AllocateOutputs() function in tensorflow::CudnnRNNForwardOp |