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

  /external/tensorflow/tensorflow/core/framework/
dataset.h 277 std::function<Allocator*(AllocatorAttributes)> allocator_getter = nullptr; member in struct:tensorflow::IteratorContext::Params
305 return params_.allocator_getter(attrs);
  /external/tensorflow/tensorflow/core/kernels/data/
iterator_ops.cc 164 params.allocator_getter = [device](AllocatorAttributes attrs) {
613 params.allocator_getter = [device](AllocatorAttributes attrs) {
    [all...]

Completed in 390 milliseconds