Home
last modified time | relevance | path

Searched refs:VariantArguments (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.h124 using VariantArguments = absl::variant<absl::Span<const ShapedBuffer* const>, variable
128 VariantArguments arguments, HloExecutionProfile* hlo_execution_profile);
160 VariantArguments arguments,
166 VariantArguments arguments,
Dgpu_executable.cc303 VariantArguments arguments, in BufferForAllocation()
375 VariantArguments arguments, in GenerateBufferAllocations()
417 const ServiceExecutableRunOptions* run_options, VariantArguments arguments, in ExecuteAsyncOnStreamImpl()