| /external/flatbuffers/grpc/examples/ts/greeter/src/ |
| D | greeter_grpc.js | 8 function serialize_models_HelloReply(buffer_args) { argument
9 if (!(buffer_args instanceof models_HelloReply)) {
12 return Buffer.from(buffer_args.serialize());
20 function serialize_models_HelloRequest(buffer_args) { argument
21 if (!(buffer_args instanceof models_HelloRequest)) {
24 return Buffer.from(buffer_args.serialize());
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | kernel_thunk.cc | 73 se::Stream* stream, absl::Span<const se::DeviceMemoryBase> buffer_args) { in PrintBufferContents() argument
75 for (const se::DeviceMemoryBase& buf : buffer_args) { in PrintBufferContents()
105 absl::InlinedVector<se::DeviceMemoryBase, 4> buffer_args; in ExecuteOnStream() local
111 buffer_args.push_back(buf); in ExecuteOnStream()
115 PrintBufferContents(params.stream, buffer_args); in ExecuteOnStream()
118 return ExecuteKernelOnStream(*kernel, buffer_args, launch_dimensions, in ExecuteOnStream()
|
| D | jitrt_custom_calls.cc | 446 absl::InlinedVector<se::DeviceMemoryBase, 4> buffer_args; in operator ()() local
447 buffer_args.reserve(args.size()); in operator ()()
454 buffer_args.emplace_back(GetDeviceAddress(*memref)); in operator ()()
462 buffer_args.emplace_back(GetDeviceAddress(*strided)); in operator ()()
471 ExecuteKernelOnStream(*kernel, buffer_args, launch_dimensions, stream); in operator ()()
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | cuda_codegen.cpp | 885 HalfChecker halfChecker(buffer_args()); in Initialize()
915 const std::vector<BufferArg> buffer_args = this->buffer_args(); in Initialize() local
916 for (size_t i = 0; i < buffer_args.size(); i++) { in Initialize()
920 const BufferArg& buffer_arg = buffer_args[i]; in Initialize()
1023 for (const size_t i : c10::irange(buffer_args.size())) { in Initialize()
1024 if (vars_in_extents.count(buffer_args[i].var())) { in Initialize()
1025 extents_buffer_args.push_back(buffer_args[i]); in Initialize()
1073 auto const& buffer_args = this->buffer_args(); in call_with_numel() local
1083 std::vector<void*> ptr_to_args(buffer_args.size()); in call_with_numel()
1084 for (size_t i = 0; i < buffer_args.size(); i++) { in call_with_numel()
[all …]
|
| D | cpp_codegen.cpp | 359 const std::vector<BufferArg>& buffer_args, in CppCodeGen() argument
362 : CodeGen(std::move(stmt), buffer_args, device, kernel_func_name) { in CppCodeGen()
374 const std::vector<BufferArg> buffer_args = this->buffer_args(); in init() local
375 for (size_t i = 0; i < buffer_args.size(); i++) { in init()
379 const BufferArg& buffer_arg = buffer_args[i]; in init()
|
| D | codegen.h | 25 std::vector<BufferArg> buffer_args,
47 std::vector<BufferArg>& buffer_args() { in buffer_args() function
51 const std::vector<BufferArg>& buffer_args() const { in buffer_args() function
|
| D | block_codegen.h | 110 const std::vector<BufferArg>& buffer_args,
113 : CodeGen(std::move(stmt), buffer_args, device, kernel_func_name) { in CodeGen() argument
|
| D | eval.h | 138 const std::vector<BufferArg>& buffer_args,
176 ExprEval(const ExprHandle& expr, const std::vector<BufferArg>& buffer_args) in ExprEval() argument
178 std::vector<BufferArg> buffer_args_extended = buffer_args; in ExprEval()
|
| D | cuda_codegen.h | 210 const std::vector<BufferArg>& buffer_args,
213 : CodeGen(std::move(stmt), buffer_args, device, kernel_func_name) { in CodeGen() argument
|
| D | cpp_codegen.h | 68 const std::vector<BufferArg>& buffer_args,
|
| D | codegen.cpp | 12 std::vector<BufferArg> buffer_args, in CodeGen() argument
16 buffer_args_(std::move(buffer_args)), in CodeGen()
|
| D | eval.cpp | 1226 const std::vector<BufferArg>& buffer_args, in SimpleIREvaluator() argument
1229 : CodeGen(std::move(stmt), buffer_args, device, kernel_func_name) { in SimpleIREvaluator()
1239 auto const& bufferArg = buffer_args()[i]; in call()
1247 if (args.size() != buffer_args().size()) { in call_raw()
1251 bindArg(buffer_args()[i], args[i]); in call_raw()
|
| D | tensorexpr_init.cpp | 859 if (py::len(values) != self.buffer_args().size()) { in initTensorExprBindings()
864 const auto& bufArg = self.buffer_args()[i]; in initTensorExprBindings()
|
| D | llvm_codegen.cpp | 430 auto& buf_args = buffer_args(); in call()
|
| /external/angle/src/libANGLE/renderer/vulkan/ |
| D | gen_vk_format_table.py | 195 def buffer_args(format): function
219 args.update(buffer_args(buffers[0]))
224 buffer_struct_template.format(**buffer_args(i)) for i in buffers),
|
| /external/angle/src/libANGLE/renderer/wgpu/ |
| D | gen_wgpu_format_table.py | 194 def buffer_args(format): function
217 args.update(buffer_args(buffers[0]))
222 buffer_struct_template.format(**buffer_args(i)) for i in buffers))
|
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_conv.cpp | 113 std::vector<te::CodeGen::BufferArg> buffer_args = { in TEST() local
128 te::LLVMCodeGen cg(loop.root_stmt(), buffer_args); in TEST()
|
| /external/pytorch/torch/csrc/jit/mobile/nnc/ |
| D | context.cpp | 215 auto buffer_args = state.preallocations_.size(); in init_execution_state() local
219 input_args + sym_shape_args + output_args + param_args + buffer_args); in init_execution_state()
|