Home
last modified time | relevance | path

Searched defs:mask_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
H A Dbert_nl_classifier.cc73 auto* mask_tensor = in Preprocess() local
208 const auto& mask_tensor = *FindTensorByName(input_tensors, input_tensor_metadatas, in InitializeFromMetadata() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/
H A Dbert_question_answerer.cc171 TfLiteTensor* mask_tensor = in Preprocess() local
/aosp_15_r20/external/pytorch/torch/_prims_common/
H A D__init__.py1881 def mask_tensor(mask: TensorLikeType, t: TensorLikeType): function