Home
last modified time | relevance | path

Searched refs:slice_dim (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dscatter_nd_op.cc520 const int64 slice_dim = in ValidateUpdateShape() local
531 ", slice_dim: ", slice_dim, ", and batch_dim: ", batch_dim); in ValidateUpdateShape()
535 if (params_shape.dims() < slice_dim + (updates.dims() - batch_dim)) { in ValidateUpdateShape()
538 if (updates.dims() != batch_dim + params_shape.dims() - slice_dim) { in ValidateUpdateShape()
546 params_shape.dim_size(d + slice_dim)) { in ValidateUpdateShape()
556 int64* slice_dim, Index* num_updates, in PrepareAndValidateInputs() argument
598 *slice_dim = (indices_shape.dims() > 1) in PrepareAndValidateInputs()
608 for (int64 i = *slice_dim; i < total_nd; ++i) { in PrepareAndValidateInputs()
620 const int64 safe_slice_dim = (*slice_dim < 1) ? 1 : *slice_dim; in PrepareAndValidateInputs()
665 int64 slice_dim; in DoScatterNd() local
[all …]
/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py170 slice_dim = None
174 if slice_dim is not None:
178 "or {}.".format(shape, self.full_shape, i, slice_dim))
179 slice_dim = i
181 return slice_dim
720 slice_dim, num_slices = _get_slice_dim_and_num_slices(partitions)
740 slice_dim
2440 for slice_dim, num_slices in enumerate(slicing):
2445 slice_dim = 0
2447 return slice_dim, num_slices
[all …]
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc978 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim) { in TFE_Py_TensorShapeSlice() argument
987 if (slice_dim < 0) { in TFE_Py_TensorShapeSlice()
992 slice_dim) in TFE_Py_TensorShapeSlice()
1021 if (slice_dim >= num_dims) { in TFE_Py_TensorShapeSlice()
1024 tensorflow::strings::StrCat("Slice dimension (", slice_dim, in TFE_Py_TensorShapeSlice()
1031 int64_t dim = TFE_TensorHandleDim(handle, slice_dim, status.get()); in TFE_Py_TensorShapeSlice()
Dpywrap_tfe.h227 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim);
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h629 const int slice_dim = values_inserted_in_slice[slice_index]++; in Split() local
630 output_values_t[slice_index](slice_dim) = input_values_t(i); in Split()
633 output_indices_t[slice_index](slice_dim, j) = in Split()