Searched defs:input_buffers (Results 1 – 8 of 8) sorted by relevance
234 auto* const input_buffers = static_cast<InputBuffers*>(callback_private_data); in ReleaseInputBuffer() local277 InputBuffers input_buffers; in main() local
66 InputBuffers input_buffers; in LLVMFuzzerTestOneInput() local
280 auto input_buffers = absl::make_unique<InputBuffers>( in BuildComputationInputs() local397 se::Stream* stream, int device_ordinal, InputBuffers* input_buffers, in AllocateOutputTensors()
152 xla::ShapeTree<xla::MaybeOwningDeviceMemory> input_buffers( in BuildInputBuffers() local
110 ShapeTree<se::DeviceMemoryBase> input_buffers = shaped_buffer.buffers(); in RunAndCheck() local
553 std::vector<std::vector<absl::Span<T>>> input_buffers; in DoAllReduce() local765 int32_t num_buffers, void** input_buffers, void** output_buffers) { in __xla_cpu_runtime_AllReduce()
1239 llvm::Value* input_buffers = in HandleAllReduceMultipleReplica() local1316 llvm::Value* input_buffers = in HandleAllToAll() local
3259 auto input_buffers = fusion.getInputBuffers(); in BuildFusedInitializerThunk() local