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 | 126 SmallVector<Value, 4> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local 149 SmallVector<Value, 2> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local 179 SmallVector<Value, 2> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local 422 SmallVector<Value, 4> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local
|
/external/angle/src/libANGLE/renderer/vulkan/ |
D | gen_vk_format_table.py | 191 def buffer_args(format): function
|