Searched defs:slice_dim (Results 1 – 8 of 8) sorted by relevance
24 TensorSlicer::TensorSlicer(const mli_tensor* full_tensor, int slice_dim, in TensorSlicer()
779 const int64 slice_dim = in ValidateUpdateShape() local820 int64* slice_dim, Index* num_updates, in PrepareAndValidateInputs()905 int64 slice_dim; in DoScatterNd() local
550 const int slice_dim = values_inserted_in_slice[slice_index]++; in Split() local
1016 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim) { in TFE_Py_TensorShapeSlice()
1031 [](const py::handle& tensors, int slice_dim) { in PYBIND11_MODULE()1036 int slice_dim) { in PYBIND11_MODULE()
942 auto slice_dim = in PartitionBaseCase() local1008 auto slice_dim = in PartitionBaseCase() local
1262 int64 slice_dim, int64 k) { in SliceFirstK()
2572 def _iter_slices(full_shape, num_slices, slice_dim): argument