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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
one_hot_op.cc 38 const int indices_dims = indices_shape.dims(); variable
39 const int output_dims = indices_dims + 1;
56 const int axis = (axis_ == -1) ? indices_dims : axis_;
  /external/tensorflow/tensorflow/compiler/tf2xla/lib/
scatter.cc 47 gtl::ArraySlice<int64> indices_dims = local
56 TF_RET_CHECK(!indices_dims.empty());
57 num_index_dims = indices_dims.back();
65 indices_dims.pop_back();
69 for (int64 dim : indices_dims) {
  /external/tensorflow/tensorflow/core/kernels/
one_hot_op.cc 57 const int indices_dims = indices_shape.dims(); variable
58 const int output_dims = indices_dims + 1;
75 const int axis = (axis_ == -1) ? indices_dims : axis_;
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_helpers.cc 233 const int indices_dims = indices_shape.dims(); local
234 const int output_dims = indices_dims + 1;

Completed in 266 milliseconds