Searched refs:RegisterBuffer (Results 1 – 5 of 5) sorted by relevance
| /external/gemmlowp/internal/ |
| D | output.h | 66 RegisterBuffer<std::int32_t, Size>> {
67 using InputType = RegisterBuffer<std::int32_t, Size>;
68 using OutputType = RegisterBuffer<std::int32_t, Size>;
123 RegisterBuffer<std::int32_t, Size>> {
124 typedef RegisterBuffer<std::int32_t, Size> InputType;
125 typedef RegisterBuffer<std::int32_t, Size> OutputType;
151 RegisterBuffer<std::int32_t, Size>> {
152 typedef RegisterBuffer<std::int32_t, Size> InputType;
153 typedef RegisterBuffer<std::int32_t, Size> OutputType;
185 RegisterBuffer<std::int32_t, Size>> {
[all …]
|
| D | simd_wrappers.h | 45 struct RegisterBuffer { struct
67 using BufferType = RegisterBuffer<ScalarType, kScalarCount>; argument
492 using RegBufferInt32 = RegisterBuffer<std::int32_t, N>;
494 using RegBufferInt16 = RegisterBuffer<std::int16_t, N>;
496 using RegBufferUint8 = RegisterBuffer<std::uint8_t, N>;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_allocations.h | 44 void RegisterBuffer(BufferAllocation::Index index,
|
| D | gpu_executable.cc | 269 buffer_allocations_builder.RegisterBuffer(i, buffer); in ExecuteOnStream()
273 buffer_allocations_builder.RegisterBuffer(i, FindOrDie(*globals, i)); in ExecuteOnStream()
|
| D | buffer_allocations.cc | 35 void BufferAllocations::Builder::RegisterBuffer(BufferAllocation::Index index, in RegisterBuffer() function in xla::gpu::BufferAllocations::Builder
|