Searched refs:output_shape_vec (Results 1 – 4 of 4) sorted by relevance
111 auto output_shape_vec = output_shape->vec<int64>(); in Compute() local112 output_shape_vec(0) = num_sparse_tensors; in Compute()114 output_shape_vec(j) = input.dim_size(j); in Compute()117 output_shape_vec(j + input_dims_to_stack) = element_shape_vec(j); in Compute()146 output_shape_vec(j + input_dims_to_stack) = std::max( in Compute()147 output_shape_vec(j + input_dims_to_stack), element_shape_vec(j)); in Compute()
88 auto output_shape_vec = output_shape.flat<Index>(); in Compute() local90 OP_REQUIRES_OK(c, TensorShapeUtils::MakeShape(output_shape_vec.data(), in Compute()91 output_shape_vec.size(), in Compute()
147 auto output_shape_vec = result_shape->vec<int64>(); in Reshape() local149 output_shape_vec(j) = output_shape.dim_size(j); in Reshape()
292 auto output_shape_vec = output_shape_t.vec<int32>(); in Compute() local293 const int64 output_height = internal::SubtleMustCopy(output_shape_vec(0)); in Compute()294 const int64 output_width = internal::SubtleMustCopy(output_shape_vec(1)); in Compute()