Searched defs:input_buffers (Results 1 – 7 of 7) sorted by relevance
66 InputBuffers input_buffers; in LLVMFuzzerTestOneInput() local
234 auto* const input_buffers = static_cast<InputBuffers*>(callback_private_data); in ReleaseInputBuffer() local277 InputBuffers input_buffers; in main() local
281 auto input_buffers = absl::make_unique<InputBuffers>( in BuildComputationInputs() local398 se::Stream* stream, int device_ordinal, InputBuffers* input_buffers, in AllocateOutputTensors()
107 ShapeTree<se::DeviceMemoryBase> input_buffers = shaped_buffer.buffers(); in RunAndCheck() local
484 std::vector<std::vector<absl::Span<T>>> input_buffers; in DoAllReduce() local664 void** input_buffers, void** output_buffers) { in __xla_cpu_runtime_AllReduce()
1165 llvm::Value* input_buffers = in HandleAllReduceMultipleReplica() local1237 llvm::Value* input_buffers = in HandleAllToAll() local
3707 auto input_buffers = fusion.getInputBuffers(); in BuildFusedInitializerThunkForMlir() local