Searched refs:AllocateOutputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_partition_op_gpu.cu.cc | 211 void AllocateOutputs(OpKernelContext* c, const Tensor* data, in AllocateOutputs() function in tensorflow::DynamicPartitionOpGPU
255 this->AllocateOutputs(c, &data, &partitions, &partition_count, &outputs, in ComputeAsync()
307 this->AllocateOutputs(c, &data, &partitions, &cpu_tensor, &outputs, done); in ComputeAsync()
|
| D | cudnn_rnn_ops.cc | 1301 OP_REQUIRES_OK(context, AllocateOutputs(context, model_shapes, &output, in ComputeAndReturnAlgorithm()
1359 Status AllocateOutputs(OpKernelContext* context, in AllocateOutputs() function in tensorflow::CudnnRNNForwardOp
1702 AllocateOutputs(context, model_shapes, params->shape(), in ComputeImpl()
1800 Status AllocateOutputs(OpKernelContext* context, in AllocateOutputs() function in tensorflow::CudnnRNNBackwardOp
|