HomeSort by relevance Sort by last modified time
    Searched refs:dim_axis (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/
reduce_slice_ops.cc 30 DimensionHandle dim_axis = c->UnknownDim(); local
45 TF_RETURN_IF_ERROR(c->Subtract(raw_dim_axis, 1, &dim_axis));
49 dim_axis = c->Dim(c->input(1), 0);
58 TF_RETURN_IF_ERROR(c->ReplaceDim(handle, axis, dim_axis, &handle));
  /external/tensorflow/tensorflow/python/ops/
nn_ops.py     [all...]

Completed in 54 milliseconds