Home
last modified time | relevance | path

Searched refs:tensor_list_index (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dlower_static_tensor_list.cc1144 const llvm::SmallSet<int, 4> &tensor_list_index, in UpdateTensorListTypes() argument
1150 if (tensor_list_index.count(i)) { in UpdateTensorListTypes()
1170 const llvm::SmallSet<int, 4> &tensor_list_index, ArrayRef<Type> types, in ChangeVariantToUnrankedTensorType() argument
1174 if (tensor_list_index.count(i)) { in ChangeVariantToUnrankedTensorType()
1336 const llvm::SmallSet<int, 4> &tensor_list_index, in UpdateTensorListResultTypesForIf() argument
1343 if (!tensor_list_index.count(i)) { in UpdateTensorListResultTypesForIf()
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_list_dict.py1106 def tensor_list_index(): function
1112 self.checkScript(tensor_list_index, ())