Home
last modified time | relevance | path

Searched refs:existing_dims_size (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc138 const int existing_dims_size = static_cast<int>(existing_dims.size()); in Compile() local
139 std::vector<int64> new_shape(existing_dims_size); in Compile()
151 dim = std::min<int32>(dim, existing_dims_size); in Compile()
171 int existing_dims_size = input_shape.dims(); in Compile() local
184 dim = existing_dims_size + dim; in Compile()
190 for (int i = 0; i < existing_dims_size; ++i) { in Compile()
/external/tensorflow/tensorflow/core/kernels/
Dshape_ops.h156 const int existing_dims_size = static_cast<int>(existing_dims.size()); in Compute() local
157 std::vector<int64> new_shape(existing_dims_size); in Compute()
169 dim = std::min<Tdim>(dim, existing_dims_size); in Compute()
201 const int existing_dims_size = static_cast<int>(existing_dims.size()); in Compute() local
215 dim = existing_dims_size + dim; in Compute()
221 for (int i = 0; i < existing_dims_size; ++i) { in Compute()