Searched defs:output_buffers (Results 1 – 12 of 12) sorted by relevance
72 std::vector<std::vector<std::unique_ptr<PjRtBuffer>>> output_buffers; in Execute() local94 std::vector<std::vector<std::unique_ptr<PjRtBuffer>>> output_buffers; in ExecuteShardedOnLocalDevices() local
889 std::vector<std::vector<std::unique_ptr<xla::PjRtBuffer>>> output_buffers; in Call() local
39 ShapeTree<std::unique_ptr<OutfeedBuffer>>* output_buffers = in ExecuteOnStream() local
897 auto output_buffers = fusion.getOutputBuffers(); in CanEmitFusedDynamicUpdateSliceInPlaceForGpu() local
122 std::vector<Offset<Buffer>> output_buffers; in StripWeightsFromFlatbuffer() local248 std::vector<Offset<::tflite::Buffer>> output_buffers; in ReconstituteConstantTensorsIntoFlatbuffer() local
270 auto output_buffers = result_buffers.release(); in UpdateOutputVariables() local
439 auto output_buffers = in AllocateOutputTensors() local
515 std::vector<std::vector<absl::Span<T>>> output_buffers; in DoAllReduce() local686 void** input_buffers, void** output_buffers) { in __xla_cpu_runtime_AllReduce()
1169 llvm::Value* output_buffers = in HandleAllReduceMultipleReplica() local1241 llvm::Value* output_buffers = in HandleAllToAll() local
154 PointsToSet::BufferSet* output_buffers = GetBufferSet(instruction); in AnyOperandBufferForwarded() local166 PointsToSet::BufferSet* output_buffers = GetBufferSet(instruction); in AllOperandBuffersForwarded() local
757 std::vector<std::vector<std::unique_ptr<PyTpuBuffer>>> output_buffers; in ExecuteShardedOnLocalDevices() local
1309 absl::InlinedVector<std::shared_ptr<MaybeOwningCpuMemory>, 4> output_buffers; in CreateResultShapedBuffer() local