Searched refs:expanded_tensor_shape (Results 1 – 2 of 2) sorted by relevance
134 TensorShape expanded_tensor_shape(expanded_shape.vec<int64>()); in Compute() local139 shape = expanded_tensor_shape; in Compute()142 context, shape.dims() == expanded_tensor_shape.dims(), in Compute()147 "] is: ", expanded_tensor_shape.dims() - 1)); in Compute()157 expanded_tensor_shape.dim_size(j))); in Compute()
443 TensorShape expanded_tensor_shape(expanded_shape_t); in Compute() local447 shapes_to_concat.push_back(std::move(expanded_tensor_shape)); in Compute()