Searched refs:TensorOptList (Results 1 – 6 of 6) sorted by relevance
| /external/executorch/kernels/portable/cpu/util/ |
| D | advanced_index_util.h | 17 using TensorOptList = exec_aten::ArrayRef<exec_aten::optional<Tensor>>; variable
24 bool check_index_args(const Tensor& in, TensorOptList indices, Tensor& out);
39 size_t count_index_blocks(TensorOptList indices);
50 TensorOptList indices,
57 size_t get_indices_broadcast_ndim(TensorOptList indices);
63 size_t get_num_indexed_dims(TensorOptList indices);
68 size_t get_num_null_indices(TensorOptList indices);
73 size_t get_num_leading_null_indices(TensorOptList indices);
80 TensorOptList indices,
91 TensorOptList indices,
[all …]
|
| D | advanced_index_util.cpp | 16 using TensorOptList = exec_aten::ArrayRef<exec_aten::optional<Tensor>>; typedef
20 bool check_indices_dtypes(TensorOptList indices) { in check_indices_dtypes()
42 bool check_mask_indices(const Tensor& in, TensorOptList indices) { in check_mask_indices()
141 bool check_index_args(const Tensor& in, TensorOptList indices, Tensor& out) { in check_index_args()
150 size_t count_index_blocks(TensorOptList indices) { in count_index_blocks()
167 TensorOptList indices, in get_indices_broadcast_shape()
211 size_t get_indices_broadcast_ndim(TensorOptList indices) { in get_indices_broadcast_ndim()
230 size_t get_num_indexed_dims(TensorOptList indices) { in get_num_indexed_dims()
245 size_t get_num_null_indices(TensorOptList indices) { in get_num_null_indices()
255 size_t get_num_leading_null_indices(TensorOptList indices) { in get_num_leading_null_indices()
[all …]
|
| /external/executorch/exir/dialects/edge/arg/ |
| D | type.py | 14 TensorOptList = "Tensor?[]" variable in ArgType
25 return self in [ArgType.TensorList, ArgType.TensorOptList]
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_index.cpp | 23 using TensorOptList = exec_aten::ArrayRef<exec_aten::optional<Tensor>>; typedef
28 TensorOptList indices, in index_Tensor_out()
|
| /external/executorch/exir/dialects/edge/dtype/ |
| D | utils.py | 24 if sample.type == ArgType.TensorList or sample.type == ArgType.TensorOptList:
|
| /external/executorch/exir/dialects/edge/op/ |
| D | sample_input.py | 618 ArgType.TensorOptList,
634 ArgType.TensorOptList,
|