Home
last modified time | relevance | path

Searched defs:buffer_args (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/flatbuffers/tests/
H A Dmonster_test_grpc.js8 function serialize_MyGame_Example_Stat(buffer_args) { argument
20 function serialize_MyGame_Example_Monster(buffer_args) { argument
/aosp_15_r20/external/flatbuffers/grpc/examples/ts/greeter/src/
H A Dgreeter_grpc.js8 function serialize_models_HelloReply(buffer_args) { argument
20 function serialize_models_HelloRequest(buffer_args) { argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dkernel_thunk.cc73 se::Stream* stream, absl::Span<const se::DeviceMemoryBase> buffer_args) { in PrintBufferContents()
105 absl::InlinedVector<se::DeviceMemoryBase, 4> buffer_args; in ExecuteOnStream() local
H A Djitrt_custom_calls.cc446 absl::InlinedVector<se::DeviceMemoryBase, 4> buffer_args; in operator ()() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcodegen.h47 std::vector<BufferArg>& buffer_args() { in buffer_args() function
51 const std::vector<BufferArg>& buffer_args() const { in buffer_args() function
H A Dcuda_codegen.cpp915 const std::vector<BufferArg> buffer_args = this->buffer_args(); in Initialize() local
1073 auto const& buffer_args = this->buffer_args(); in call_with_numel() local
1116 auto const& buffer_args = this->buffer_args(); in call_raw() local
1240 auto const& buffer_args = this->buffer_args(); in call() local
H A Dcpp_codegen.cpp359 const std::vector<BufferArg>& buffer_args, in CppCodeGen()
374 const std::vector<BufferArg> buffer_args = this->buffer_args(); in init() local
H A Dblock_codegen.h113 : CodeGen(std::move(stmt), buffer_args, device, kernel_func_name) { in CodeGen() argument
H A Dcuda_codegen.h213 : CodeGen(std::move(stmt), buffer_args, device, kernel_func_name) { in CodeGen() argument
H A Deval.h176 ExprEval(const ExprHandle& expr, const std::vector<BufferArg>& buffer_args) in ExprEval()
H A Dcodegen.cpp12 std::vector<BufferArg> buffer_args, in CodeGen()
H A Deval.cpp1226 const std::vector<BufferArg>& buffer_args, in SimpleIREvaluator()
/aosp_15_r20/external/angle/src/libANGLE/renderer/vulkan/
H A Dgen_vk_format_table.py195 def buffer_args(format): function
/aosp_15_r20/external/angle/src/libANGLE/renderer/wgpu/
H A Dgen_wgpu_format_table.py194 def buffer_args(format): function
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/nnc/
H A Dcontext.cpp215 auto buffer_args = state.preallocations_.size(); in init_execution_state() local
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_conv.cpp113 std::vector<te::CodeGen::BufferArg> buffer_args = { in TEST() local