HomeSort by relevance Sort by last modified time
    Searched refs:ret_alloc_attrs (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
partitioning_utils.h 63 std::vector<AllocatorAttributes>* ret_alloc_attrs);
partitioning_utils.cc 76 std::vector<AllocatorAttributes>* ret_alloc_attrs) {
118 ret_alloc_attrs->push_back(alloc_attr);
partitioning_utils_test.cc 184 std::vector<AllocatorAttributes> ret_alloc_attrs; local
188 &arg_alloc_attrs, &ret_alloc_attrs);
194 CheckAlloc({false}, ret_alloc_attrs);
function.cc 988 AllocatorAttributes ret_alloc_attrs; local
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
function_ops.cc 370 AllocatorAttributes ret_alloc_attrs; local
372 ret_alloc_attrs.set_on_host(true);
374 opts.rets_alloc_attrs.push_back(ret_alloc_attrs);

Completed in 308 milliseconds