Searched refs:expanded_shape_t (Results 1 – 2 of 2) sorted by relevance
128 auto expanded_shape_t = expanded_shape.vec<int64>(); in Compute() local129 expanded_shape_t(0) = 1; in Compute()130 std::copy_n(&output_shape_t(0), rank, &expanded_shape_t(1)); in Compute()
431 auto expanded_shape_t = expanded_shape.vec<int64>(); in Compute() local436 expanded_shape_t(0) = 1; in Compute()440 expanded_shape_t(i + 1) = output_shape[i]; in Compute()442 TensorShape expanded_tensor_shape(expanded_shape_t); in Compute()