Searched refs:input_allocators (Results 1 – 1 of 1) sorted by relevance
459 std::deque<WriteIntoXlaBufferAllocator> input_allocators; in CallTfKernel() local487 input_allocators.emplace_back(buffers[i - constant_offset], input_size, in CallTfKernel()489 input_tensors.emplace_back(&input_allocators[i], dt, shape); in CallTfKernel()