Searched defs:output_buffers (Results 1 – 14 of 14) sorted by relevance
39 ShapeTree<std::unique_ptr<OutfeedBuffer>>* output_buffers = in ExecuteOnStream() local
638 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
97 std::vector<std::vector<std::unique_ptr<PjRtBuffer>>> output_buffers; in ExecuteInternal() local195 std::vector<std::vector<std::unique_ptr<PjRtBuffer>>> output_buffers; in ExecuteShardedOnLocalDevicesInternal() local
508 std::vector<std::vector<std::unique_ptr<xla::PjRtBuffer>>> output_buffers; in Call() local
997 std::vector<std::vector<std::unique_ptr<xla::PjRtBuffer>>> output_buffers; in Call() local
270 auto output_buffers = result_buffers.release(); in UpdateOutputVariables() local
438 auto output_buffers = in AllocateOutputTensors() local
283 let output_buffers = (0..NUM_OUTPUT_BUFFERS) in decode_h264_generic() localVariable
554 std::vector<std::vector<absl::Span<T>>> output_buffers; in DoAllReduce() local765 int32_t num_buffers, void** input_buffers, void** output_buffers) { in __xla_cpu_runtime_AllReduce()
1241 llvm::Value* output_buffers = in HandleAllReduceMultipleReplica() local1318 llvm::Value* output_buffers = in HandleAllToAll() local
159 PointsToSet::BufferSet* output_buffers = GetBufferSet(instruction); in AnyOperandBufferForwarded() local171 PointsToSet::BufferSet* output_buffers = GetBufferSet(instruction); in AllOperandBuffersForwarded() local
767 std::vector<std::vector<std::unique_ptr<PyTpuBuffer>>> output_buffers; in ExecuteShardedOnLocalDevices() local
1275 absl::InlinedVector<std::shared_ptr<MaybeOwningCpuMemory>, 4> output_buffers; in CreateResultShapedBuffer() local