Searched defs:indices_list (Results 1 – 6 of 6) sorted by relevance
401 def _validate_indices_list(indices_list, formatted): argument424 def _locate_elements_in_line(line, indices_list, ref_indices): argument
36 c10::List<std::optional<at::Tensor>> indices_list(indices); in op_index_tensor_out() local
37 c10::List<std::optional<at::Tensor>> indices_list(indices); in op_index_put_out() local
349 VmapDimVector compute_indexed_shape(const Tensor &src, TensorList indices_list) in compute_indexed_shape()
553 AdvancedIndex::AdvancedIndex(const Tensor& src, TensorList indices_list) in AdvancedIndex()
2117 Slice<mpy::handle> indices_list; in index() local