Searched defs:buffer_args (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | kernel_thunk.cc | 71 se::Stream* stream, absl::Span<const se::DeviceMemoryBase> buffer_args) { in PrintBufferContents()
103 absl::InlinedVector<se::DeviceMemoryBase, 4> buffer_args; in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_to_lhlo.cc | 145 SmallVector<Value, 4> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local
168 SmallVector<Value, 2> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local
198 SmallVector<Value, 2> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local
268 SmallVector<Value, 4> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local
|
| /external/angle/src/libANGLE/renderer/vulkan/ |
| D | gen_vk_format_table.py | 190 def buffer_args(format): function
|