Searched defs:slice_dim (Results 1 – 7 of 7) sorted by relevance
779 const int64_t slice_dim = in ValidateUpdateShape() local820 int64* slice_dim, Index* num_updates, in PrepareAndValidateInputs()905 int64_t slice_dim; in DoScatterNd() local
554 const int slice_dim = values_inserted_in_slice[slice_index]++; in Split() local
1114 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim) { in TFE_Py_TensorShapeSlice()
1144 [](const py::handle& tensors, int slice_dim) { in PYBIND11_MODULE()1149 int slice_dim) { in PYBIND11_MODULE()
2574 def _iter_slices(full_shape, num_slices, slice_dim): argument
1028 auto slice_dim = in PartitionBaseCase() local1096 auto slice_dim = in PartitionBaseCase() local
1308 int64_t slice_dim, int64_t k) { in SliceFirstK()