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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
shape_op.cc 132 auto existing_dims = input_shape.dim_sizes(); variable
134 const int existing_dims_size = static_cast<int>(existing_dims.size());
137 new_shape[i] = existing_dims[i];
143 dim += existing_dims.size() + 1;
165 auto existing_dims = input_shape.dim_sizes(); variable
186 auto existing_dim = existing_dims[i];
  /external/tensorflow/tensorflow/core/kernels/
shape_ops.h 164 auto existing_dims = ctx->input(0).shape().dim_sizes(); variable
166 const int existing_dims_size = static_cast<int>(existing_dims.size());
169 new_shape[i] = existing_dims[i];
175 dim += existing_dims.size() + 1;
210 auto existing_dims = ctx->input(0).shape().dim_sizes(); variable
211 const int existing_dims_size = static_cast<int>(existing_dims.size());
232 auto existing_dim = existing_dims[i];

Completed in 81 milliseconds