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

  /external/tensorflow/tensorflow/core/common_runtime/
partitioning_utils.h 62 std::vector<AllocatorAttributes>* arg_alloc_attrs,
partitioning_utils.cc 75 std::vector<AllocatorAttributes>* arg_alloc_attrs,
107 arg_alloc_attrs->push_back(alloc_attr);
partitioning_utils_test.cc 183 std::vector<AllocatorAttributes> arg_alloc_attrs; local
188 &arg_alloc_attrs, &ret_alloc_attrs);
193 CheckAlloc({false}, arg_alloc_attrs);
function.cc 981 AllocatorAttributes arg_alloc_attrs; local
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
function_ops.cc 363 AllocatorAttributes arg_alloc_attrs; local
365 arg_alloc_attrs.set_on_host(true);
367 opts.args_alloc_attrs.push_back(arg_alloc_attrs);

Completed in 137 milliseconds