Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dshape_ops.h154 auto existing_dims = ctx->input(0).shape().dim_sizes(); in Compute() local
156 const int existing_dims_size = static_cast<int>(existing_dims.size()); in Compute()
159 new_shape[i] = existing_dims[i]; in Compute()
165 dim += existing_dims.size() + 1; in Compute()
200 auto existing_dims = ctx->input(0).shape().dim_sizes(); in Compute() local
201 const int existing_dims_size = static_cast<int>(existing_dims.size()); in Compute()
222 auto existing_dim = existing_dims[i]; in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc136 auto existing_dims = input_shape.dim_sizes(); in Compile() local
138 const int existing_dims_size = static_cast<int>(existing_dims.size()); in Compile()
141 new_shape[i] = existing_dims[i]; in Compile()
147 dim += existing_dims.size() + 1; in Compile()
170 auto existing_dims = input_shape.dim_sizes(); in Compile() local
191 auto existing_dim = existing_dims[i]; in Compile()