Searched defs:slice_dim (Results 1 – 4 of 4) sorted by relevance
520 const int64 slice_dim = in ValidateUpdateShape() local556 int64* slice_dim, Index* num_updates, in PrepareAndValidateInputs()665 int64 slice_dim; in DoScatterNd() local
978 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim) { in TFE_Py_TensorShapeSlice()
629 const int slice_dim = values_inserted_in_slice[slice_index]++; in Split() local
2450 def _iter_slices(full_shape, num_slices, slice_dim): argument