Searched defs:shape_dims (Results 1 – 4 of 4) sorted by relevance
55 size_t shape_dims = 1 + (data[data_ix++] % kMaxShapeDims); in FuzzImpl() local
105 auto shape_dims = AsInt64Slice(blocks_shape.dimensions()); in DiagonalBlocks() local
613 auto* shape_dims = shape->mutable_dims(); in ExtendShape() local624 std::vector<int>& shape_dims = *shape->mutable_dims(); in UnextendShape() local
139 const std::vector<int32> shape_dims = {static_cast<int32>(vals.size())}; in MakeConstNodeDef() local