Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/core/framework/ | |
op_kernel.cc | 217 Status OpKernelConstruction::allocate_temp(DataType type, function in class:tensorflow::OpKernelConstruction 239 // for now just do the same thing as allocate_temp 242 Status s = allocate_temp(type, shape, &persistent); 509 return allocate_temp(type, shape, out_temp, allocator_attr); 666 Status OpKernelContext::allocate_temp( function in class:tensorflow::OpKernelContext [all...] |
op_kernel.h | 249 // allocate_temp below. Some kernels need to keep tensors in between 262 Status allocate_temp(DataType type, const TensorShape& shape, 732 // allocate_temp() to allocate a new temporary buffer. 785 // 3) allocate_temp. This should be used to allocate any scratch 834 Status allocate_temp(DataType type, const TensorShape& shape, function in class:tensorflow::OpKernelContext 839 Status allocate_temp(DataType type, const TensorShape& shape, function in class:tensorflow::OpKernelContext [all...] |