Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsubgraph.h36 std::vector<int> input_ids; member
/external/XNNPACK/src/subgraph/
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
/external/perfetto/protos/perfetto/trace/android/
Dcamera_event.proto126 repeated int64 input_ids = 2; field
/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
Dbert_nl_classifier.cc109 std::vector<int> input_ids(input_tensor_length, 0); in Preprocess() local
/external/tensorflow/tensorflow/core/kernels/
Dtraining_op_helpers.h142 const std::vector<int>& input_ids) { in MaybeLockVariableInputMutexesInOrder()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
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
Dmodel_builder.cc3296 const std::vector<int>& input_ids, const std::vector<int>& output_ids, in BuildModelEnforceIO()
3387 std::vector<int> input_ids; member
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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()
/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/
Dbert_question_answerer.cc259 std::vector<int> input_ids(tokens.size()); in Preprocess() local
/external/tensorflow/tensorflow/python/eager/
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
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Drecording_tpu_driver.cc424 std::vector<int64_t> input_ids; in ExecuteProgram() local
/external/tensorflow/tensorflow/c/eager/
Dgradients.cc229 std::vector<int64_t> input_ids(inputs.size()); in RecordOperation() local
/external/tensorflow/tensorflow/c/
Dkernels_experimental.cc295 std::vector<int> input_ids(inputs, inputs + len); in TF_MaybeLockVariableInputMutexesInOrder() local
/external/libchrome-gestures/src/
Dlookahead_filter_interpreter_unittest.cc893 std::set<short> input_ids; in TEST() local
/external/perfetto/protos/perfetto/trace/
Dperfetto_trace.proto3743 repeated int64 input_ids = 2; field