Searched refs:shape_dims (Results 1 – 5 of 5) sorted by relevance
55 size_t shape_dims = 1 + (data[data_ix++] % kMaxShapeDims); in FuzzImpl() local57 TensorShape({static_cast<int64>(shape_dims)})); in FuzzImpl()60 if (data_ix + shape_dims >= size) { in FuzzImpl()67 for (i = 0; i < shape_dims; i++) { in FuzzImpl()110 for (i = last; i < shape_dims; i++) { in FuzzImpl()
194 shape_dims = []199 shape_dims.append(shape_dim)200 shape = array_ops.reshape(array_ops.stack(shape_dims), [-1])
105 auto shape_dims = AsInt64Slice(blocks_shape.dimensions()); in DiagonalBlocks() local107 std::copy(shape_dims.begin(), shape_dims.end(), last_blocks_dims.begin()); in DiagonalBlocks()
613 auto* shape_dims = shape->mutable_dims(); in ExtendShape() local614 shape_dims->insert(shape_dims->begin(), size_increase, 1); in ExtendShape()624 std::vector<int>& shape_dims = *shape->mutable_dims(); in UnextendShape() local625 shape_dims.erase(shape_dims.begin(), shape_dims.begin() + size_reduction); in UnextendShape()
139 const std::vector<int32> shape_dims = {static_cast<int32>(vals.size())}; in MakeConstNodeDef() local140 TF_EXPECT_OK(TensorShapeUtils::MakeShape(shape_dims, &shape)); in MakeConstNodeDef()