Searched refs:vector3_type (Results 1 – 1 of 1) sorted by relevance
101 auto vector3_type = RankedTensorType::get({3}, int64_type); in sliceInput() local104 DenseElementsAttr::get<int64_t>(vector3_type, {batch_idx, 0, 0}); in sliceInput()105 auto size_attr = DenseElementsAttr::get<int64_t>(vector3_type, slice_size); in sliceInput()106 auto begin = rewriter.create<TF::ConstOp>(loc, vector3_type, begin_attr); in sliceInput()107 auto size = rewriter.create<TF::ConstOp>(loc, vector3_type, size_attr); in sliceInput()