HomeSort by relevance Sort by last modified time
    Searched defs:allocate_temp (Results 1 - 2 of 2) sorted by null

  /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...]

Completed in 95 milliseconds