Home
last modified time | relevance | path

Searched defs:input_ids (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/XNNPACK/src/subgraph/
H A Dconcatenate.c509 uint32_t* input_ids, in xnn_define_concatenate_n()
649 uint32_t input_ids[2] = { input1_id, input2_id }; in xnn_define_concatenate2() local
663 uint32_t input_ids[3] = { input1_id, input2_id, input3_id }; in xnn_define_concatenate3() local
678 uint32_t input_ids[4] = { input1_id, input2_id, input3_id, input4_id }; in xnn_define_concatenate4() local
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/
H A Dio_memory.h72 int32_t input_ids[1024 * 2]; member
134 int32_t input_ids; member
H A Dio_memory.cpp96 Result<TensorInfo> input_ids = methods_meta[0]->input_tensor_meta(0); in prepare_io() local
258 Result<TensorInfo> input_ids = methods_meta[0]->input_tensor_meta(0); in prepare_io() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dgpu_model_generated.h154 const flatbuffers::Vector<int32_t> *input_ids() const { in input_ids() function
256 const flatbuffers::Vector<int32_t> *input_ids() const { in input_ids() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Dsubgraph.h36 std::vector<int> input_ids; member
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe_src.cc2366 const std::vector<int64_t>& input_ids,
2386 const std::vector<int64_t>& input_ids,
2587 const std::vector<int64_t>& input_ids,
2626 std::vector<int64_t> input_ids = MakeTensorIDList(input_tensors); local
2680 std::vector<int64_t> input_ids = MakeTensorIDList(input_tensors); local
2731 std::vector<int64_t> input_ids = MakeTensorIDList(input_tensors); local
3227 std::vector<int64_t> input_ids = MakeTensorIDList(inputs); local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emission_utils.cc755 const HloComputation* fused_computation, std::vector<int64_t> input_ids) { in FilterInputsForShmemTranspose()
772 std::vector<int64_t> input_ids) { in ShmemTransposeSupportedForInputs()
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/serialization/
H A Dserializer.cpp79 std::vector<uint32_t> input_ids, in finishAndSerialize()
/aosp_15_r20/external/executorch/backends/xnnpack/runtime/
H A DXNNExecutor.cpp32 std::vector<uint32_t>&& input_ids, in initialize()
H A DXNNCompiler.cpp195 std::vector<uint32_t>& input_ids, in defineTensor()
1920 std::vector<uint32_t> input_ids; in compileModel() local
/aosp_15_r20/external/perfetto/protos/perfetto/trace/android/
H A Dcamera_event.proto126 repeated int64 input_ids = 2; field
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
H A Dbert_nl_classifier.cc109 std::vector<int> input_ids(input_tensor_length, 0); in Preprocess() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_op_helpers.h142 const std::vector<int>& input_ids) { in MaybeLockVariableInputMutexesInOrder()
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/
H A Dxnnpack_graph_builder.cpp133 std::vector<uint32_t> input_ids; in serializedXNNGraph() local
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_loop_ordering.py323 def forward(self, input_ids, labels, position_ids): argument
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/
H A Dbert_question_answerer.cc259 std::vector<int> input_ids(tokens.size()); in Preprocess() local
/aosp_15_r20/external/pytorch/test/onnx/dynamo/
H A Dtest_dynamo_with_onnxruntime_backend.py608 def forward(self, input_ids, attention_mask, position_ids): argument
/aosp_15_r20/prebuilts/runtime/mainline/platform/sdk/include_gen/external/perfetto/perfetto_protos_perfetto_trace_android_lite_gen_headers/gen/external/perfetto/protos/perfetto/trace/android/
H A Dcamera_event.pb.h2179 inline ::int64_t AndroidCameraSessionStats_CameraGraph_CameraNode::input_ids(int index) const { in input_ids() function
2199 AndroidCameraSessionStats_CameraGraph_CameraNode::input_ids() const { in input_ids() function
/aosp_15_r20/out/soong/.intermediates/external/perfetto/perfetto_protos_perfetto_trace_android_lite_gen_headers/gen/external/perfetto/protos/perfetto/trace/android/
Dcamera_event.pb.h2179 inline ::int64_t AndroidCameraSessionStats_CameraGraph_CameraNode::input_ids(int index) const { in input_ids() function
2199 AndroidCameraSessionStats_CameraGraph_CameraNode::input_ids() const { in input_ids() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Drecording_tpu_driver.cc424 std::vector<int64_t> input_ids; in ExecuteProgram() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dgradients.cc229 std::vector<int64_t> input_ids(inputs.size()); in RecordOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dkernels_experimental.cc295 std::vector<int> input_ids(inputs, inputs + len); in TF_MaybeLockVariableInputMutexesInOrder() local
/aosp_15_r20/external/libchrome-gestures/src/
H A Dlookahead_filter_interpreter_unittest.cc889 std::set<short> input_ids; in TEST() local
/aosp_15_r20/prebuilts/runtime/mainline/platform/sdk/include_gen/external/perfetto/perfetto_protos_perfetto_trace_android_zero_gen_headers/gen/external/perfetto/protos/perfetto/trace/android/
H A Dcamera_event.pbzero.h338 ::protozero::RepeatedFieldIterator<int64_t> input_ids() const { return GetRepeated<int64_t>(2); } in input_ids() function
/aosp_15_r20/out/soong/.intermediates/external/perfetto/perfetto_protos_perfetto_trace_android_zero_gen_headers/gen/external/perfetto/protos/perfetto/trace/android/
Dcamera_event.pbzero.h338 ::protozero::RepeatedFieldIterator<int64_t> input_ids() const { return GetRepeated<int64_t>(2); } in input_ids() function

12