Searched refs: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() argument
73 for (const se::DeviceMemoryBase& buf : buffer_args) { in PrintBufferContents()
103 absl::InlinedVector<se::DeviceMemoryBase, 4> buffer_args; in ExecuteOnStream() local
109 buffer_args.push_back(buf); in ExecuteOnStream()
113 PrintBufferContents(params.stream, buffer_args); in ExecuteOnStream()
118 return ExecuteKernelOnStream(*kernel, buffer_args, launch_dimensions, in ExecuteOnStream()
|
| /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
127 if (failed(ConvertResults(op, buffer_args, rewriter))) return failure(); in matchAndRewrite()
129 buffer_args, op->getAttrs()); in matchAndRewrite()
131 op, llvm::makeArrayRef(buffer_args).drop_front(operands.size())); in matchAndRewrite()
149 SmallVector<Value, 2> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local
150 if (failed(ConvertResults(op, buffer_args, rewriter))) return failure(); in matchAndRewrite()
159 buffer_args, op->getAttrs()); in matchAndRewrite()
165 rewriter.replaceOp(op, ArrayRef<Value>(buffer_args).slice(operands.size())); in matchAndRewrite()
179 SmallVector<Value, 2> buffer_args(operands.begin(), operands.end()); in matchAndRewrite() local
180 if (failed(ConvertResults(op, buffer_args, rewriter))) return failure(); in matchAndRewrite()
[all …]
|
| /external/angle/src/libANGLE/renderer/vulkan/ |
| D | gen_vk_format_table.py | 191 def buffer_args(format): function
212 args.update(buffer_args(buffers[0]))
217 buffer_struct_template.format(**buffer_args(i)) for i in buffers),
|