Searched defs:slice_dim (Results 1 – 10 of 10) sorted by relevance
25 const int64_t slice_dim = in ValidateScatterNdUpdateShape() local
779 int64_t* slice_dim, Index* num_updates, in PrepareAndValidateInputs()954 int64_t slice_dim; in DoScatterNd() local
202 int64_t slice_dim = new_dot_shape.dimensions_size() - (lhs_same ? 1 : 2); in TryMergeSameOperand() local
4855 int64_t slice_dim = 0; in HandleDynamicSlice() local
555 const int slice_dim = values_inserted_in_slice[slice_index]++; in Split() local
1201 [](const py::handle& tensors, int slice_dim) { in PYBIND11_MODULE()1206 int slice_dim) { in PYBIND11_MODULE()
1122 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim) { in TFE_Py_TensorShapeSlice()
2668 def _iter_slices(full_shape, num_slices, slice_dim): argument
1290 auto slice_dim = in PartitionBaseCase() local1358 auto slice_dim = in PartitionBaseCase() local
1224 int64_t slice_dim, int64_t k) { in SliceFirstK()