Searched refs:expanded_tensor_shape (Results 1 – 2 of 2) sorted by relevance
132 TensorShape expanded_tensor_shape(expanded_shape.vec<int64>()); in Compute() local137 shape = expanded_tensor_shape; in Compute()140 context, shape.dims() == expanded_tensor_shape.dims(), in Compute()145 "] is: ", expanded_tensor_shape.dims() - 1)); in Compute()155 expanded_tensor_shape.dim_size(j))); in Compute()
442 TensorShape expanded_tensor_shape(expanded_shape_t); in Compute() local446 shapes_to_concat.push_back(std::move(expanded_tensor_shape)); in Compute()