HomeSort by relevance Sort by last modified time
    Searched defs:slice_dim (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
slice_op.cc 291 // criteria matches for slice_dim: if indices for slice are 0 in all dims
292 // except slice_dim and if sizes of all the dimensions of the slice are same
293 // as the sizes of all the dimensions of the input except slice_dim, then
298 int slice_dim) {
300 if (dim != slice_dim &&
317 // Returns True if Slicing over a single dimension, and sets slice_dim
322 int* slice_dim) {
325 *slice_dim = dim;
335 int slice_dim = -1; local
342 DoesSliceShapeDifferInOnly1D(in_shape, begin, size, &slice_dim)) {
    [all...]
scatter_nd_op.cc 293 // params_shape[slice_dim:]
296 const int64 slice_dim = local
303 "params_shape[slice_dim:], got updates.shape: ",
307 ", slice_dim: ", slice_dim, ", and batch_dim: ", batch_dim);
311 if (params_shape.dims() < slice_dim + (updates.dims() - batch_dim)) {
314 if (updates.dims() != batch_dim + params_shape.dims() - slice_dim) {
322 params_shape.dim_size(d + slice_dim)) {
332 int64* slice_dim, Index* num_updates,
373 *slice_dim = (indices_shape.dims() > 1
440 int64 slice_dim; local
    [all...]
  /external/tensorflow/tensorflow/core/util/sparse/
sparse_tensor.h 548 const int slice_dim = values_inserted_in_slice[slice_index]++; local
549 output_values_t[slice_index](slice_dim) = input_values_t(i);
552 output_indices_t[slice_index](slice_dim, j) =

Completed in 100 milliseconds