Searched defs:transfer_op (Results 1 – 4 of 4) sorted by relevance
34 int number_of_threads, std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuTransferAsyncOpKernelBase()97 int number_of_threads, std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuTransferAsyncOpKernel()116 int number_of_threads, std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuTransferAsyncDynamicOrdinalOpKernel()
43 std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuOutfeedDequeueOp()87 std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuOutfeedDequeueTupleOp()
421 std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuInfeedEnqueueOp()465 std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuInfeedEnqueueTupleOp()538 std::unique_ptr<TpuTransferOpInterface> transfer_op) in InfeedEnqueuePrelinearizedBufferOp()
169 auto transfer_op = builder.create<tfrt::corert::TransferOp>( in runOnFunction() local