Searched refs:output_shape_vec (Results 1 – 7 of 7) 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()
168 absl::InlinedVector<int64, 8> output_shape_vec(input_dims + 1); in Compute()170 output_shape_vec[i] = input_shape.dim_size(i); in Compute()172 output_shape_vec[dim] = 1; in Compute()174 output_shape_vec[i] = input_shape.dim_size(i - 1); in Compute()176 TensorShape output_shape(output_shape_vec); 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()
164 auto output_shape_vec = result_shape->vec<int64>(); in ReshapeSparseTensor() local166 output_shape_vec(j) = output_shape.dim_size(j); in ReshapeSparseTensor()
155 auto output_shape_vec = output_shape.vec<int64>(); in Compute() local156 if (rank == 3) output_shape_vec(0) = batch_size; in Compute()157 output_shape_vec(row_dim) = a_shape.dim_size(row_dim); in Compute()158 output_shape_vec(row_dim + 1) = b_shape.dim_size(row_dim + 1); in Compute()174 const int64 num_output_rows = output_shape_vec(row_dim); in Compute()
293 auto output_shape_vec = output_shape_t.vec<int32>(); in Compute() local294 const int64 output_height = internal::SubtleMustCopy(output_shape_vec(0)); in Compute()295 const int64 output_width = internal::SubtleMustCopy(output_shape_vec(1)); in Compute()
2219 std::vector<Shape> output_shape_vec; in InferReduceWindowShape() local2224 output_shape_vec.push_back(cur_output_shape); in InferReduceWindowShape()2227 CHECK_EQ(output_shape_vec.size(), 1); in InferReduceWindowShape()2228 return output_shape_vec[0]; in InferReduceWindowShape()2230 return ShapeUtil::MakeTupleShape(output_shape_vec); in InferReduceWindowShape()