Home
last modified time | relevance | path

Searched refs:input_buffers (Results 1 – 9 of 9) sorted by relevance

/external/libgav1/libgav1/tests/fuzzer/
Ddecoder_fuzzer_frame_parallel.cc66 InputBuffers input_buffers; in LLVMFuzzerTestOneInput() local
79 settings.callback_private_data = &input_buffers; in LLVMFuzzerTestOneInput()
93 if (input_buffers.free_buffers.empty()) { in LLVMFuzzerTestOneInput()
98 input_buffers.free_buffers.push_back(buffer); in LLVMFuzzerTestOneInput()
100 input_buffer = input_buffers.free_buffers.front(); in LLVMFuzzerTestOneInput()
101 input_buffers.free_buffers.pop_front(); in LLVMFuzzerTestOneInput()
135 input_buffers.free_buffers.push_back(input_buffer); in LLVMFuzzerTestOneInput()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc281 auto input_buffers = absl::make_unique<InputBuffers>( in BuildComputationInputs() local
286 transfer_manager->GetByteSizeRequirement(input_buffers->buffers.shape()); in BuildComputationInputs()
287 TF_ASSIGN_OR_RETURN(*input_buffers->buffers.mutable_element({}), in BuildComputationInputs()
301 auto* in_buffer = input_buffers->buffers.mutable_element(in_index); in BuildComputationInputs()
355 input_buffers->variables = std::move(variables); in BuildComputationInputs()
356 input_buffers->variable_index = std::move(variable_index); in BuildComputationInputs()
358 return std::move(input_buffers); in BuildComputationInputs()
398 se::Stream* stream, int device_ordinal, InputBuffers* input_buffers, in AllocateOutputTensors() argument
512 const int variable_index = input_buffers->variable_index.at(input_index); in AllocateOutputTensors()
549 const int variable_index = input_buffers->variable_index.at(input_index); in AllocateOutputTensors()
[all …]
/external/libgav1/libgav1/examples/
Dgav1_decode.cc234 auto* const input_buffers = static_cast<InputBuffers*>(callback_private_data); in ReleaseInputBuffer() local
235 input_buffers->ReleaseInputBuffer( in ReleaseInputBuffer()
277 InputBuffers input_buffers; in main() local
286 settings.callback_private_data = &input_buffers; in main()
323 input_buffer = input_buffers.GetFreeBuffer(); in main()
335 input_buffers.ReleaseInputBuffer(input_buffer); in main()
342 input_buffers.ReleaseInputBuffer(input_buffer); in main()
419 input_buffers.ReleaseInputBuffer(input_buffer); in main()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc107 ShapeTree<se::DeviceMemoryBase> input_buffers = shaped_buffer.buffers(); in RunAndCheck() local
108 inputs_buffers.push_back(input_buffers); in RunAndCheck()
115 input_buffers.element(index), executor_->device_ordinal(), in RunAndCheck()
118 *device_memory = input_buffers.element(index); in RunAndCheck()
Dexhaustive_op_test_utils.h255 std::vector<ScopedShapedBuffer> input_buffers;
256 absl::c_transform(input_literals, std::back_inserter(input_buffers),
266 input_buffers, std::back_inserter(input_buffer_pointers),
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc484 std::vector<std::vector<absl::Span<T>>> input_buffers; in DoAllReduce() local
486 input_buffers.reserve(num_participants); in DoAllReduce()
495 input_buffers.emplace_back(); in DoAllReduce()
498 input_buffers.back(); in DoAllReduce()
527 input_buffers[participant_idx][buffer_idx][idx]); in DoAllReduce()
664 void** input_buffers, void** output_buffers) { in __xla_cpu_runtime_AllReduce() argument
690 input_buffers[i], xla::ShapeUtil::ByteSizeOf(subshape)); in __xla_cpu_runtime_AllReduce()
Dcpu_runtime.h178 void** input_buffers, void** output_buffers);
Dir_emitter.cc1165 llvm::Value* input_buffers = in HandleAllReduceMultipleReplica() local
1194 /*input_buffers=*/b_.CreateBitCast(input_buffers, i8_ptr_type), in HandleAllReduceMultipleReplica()
1237 llvm::Value* input_buffers = in HandleAllToAll() local
1255 /*source_buffers=*/b_.CreateBitCast(input_buffers, i8_ptr_type), in HandleAllToAll()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3707 auto input_buffers = fusion.getInputBuffers(); in BuildFusedInitializerThunkForMlir() local
3714 ir_arrays[input_buffers.size() + output_index]; in BuildFusedInitializerThunkForMlir()