Searched defs:output_buffers (Results 1 – 9 of 9) sorted by relevance
75 std::vector<std::vector<std::unique_ptr<PjRtBuffer>>> output_buffers; in PjRtExecute() local92 std::vector<std::vector<std::unique_ptr<PjRtBuffer>>> output_buffers; in Execute() local114 std::vector<std::vector<std::unique_ptr<PjRtBuffer>>> output_buffers; in ExecuteOnLocalDevices() local142 std::vector<std::vector<std::unique_ptr<PjRtBuffer>>> output_buffers; in ExecuteShardedOnLocalDevices() local
42 ShapeTree<std::unique_ptr<OutfeedBuffer>>* output_buffers = in ExecuteOnStream() local
801 auto output_buffers = fusion.getOutputBuffers(); in CanEmitFusedDynamicUpdateSliceInPlaceForGpu() local
485 std::vector<std::vector<absl::Span<T>>> output_buffers; in DoAllReduce() local664 void** input_buffers, void** output_buffers) { in __xla_cpu_runtime_AllReduce()
1167 llvm::Value* output_buffers = in HandleAllReduceMultipleReplica() local1239 llvm::Value* output_buffers = in HandleAllToAll() local
439 auto output_buffers = in AllocateOutputTensors() local
757 std::vector<std::vector<std::unique_ptr<PyTpuBuffer>>> output_buffers; in ExecuteShardedOnLocalDevices() local
151 PointsToSet::BufferSet* output_buffers = GetBufferSet(instruction); in OperandBufferForwarded() local
186 SmallVector<Value, 4> output_buffers; in matchAndRewrite() local